fix #1

Author: khizirsiddiqui

KD_Lib/KD/common/base_class.py

@@ -56,7 +56,7 @@ def __init__(
         if device == "cpu":
             self.device = torch.device("cpu")
         elif device == "cuda":
-	    if torch.cuda.is_available():
+            if torch.cuda.is_available():
                 self.device = torch.device("cuda")
             else:
                 print(

tests/test_KD_Lib.py

@@ -23,7 +23,7 @@
     ProbShift,
     LabelSmoothReg,
     DML,
-    BaseClass
+    BaseClass,
 )
 
 from KD_Lib.models import (
@@ -164,7 +164,8 @@ def test_LSTMNet():
 #   Strategy TESTS
 #
 
-def test_BaseClass()
+
+def test_BaseClass():
     teac = Shallow(hidden_size=400)
     stud = Shallow(hidden_size=100)