fix module send to cuda

norch/nn/module.py

@@ -53,8 +53,6 @@ class Module(ABC):
     def to(self, device):
         for _, _, parameter in self.parameters():
             parameter.to(device)
-
-        return self
     
     def state_dict(self):
         state = OrderedDict()

tests/test_autograd.py

@@ -9,7 +9,7 @@ class TestTensorAutograd(unittest.TestCase):
         self.device = os.environ.get('device')
         if self.device is None or self.device != 'cuda':
             self.device = 'cpu'
-        
+
         print(f"Running tests on: {self.device}")