fix unitest

AIboy996 · Apr 20, 2024 · 3980bf1 · 3980bf1
1 parent 6ad8653
commit 3980bf1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_autograd.py b/tests/test_autograd.py
@@ -253,7 +253,7 @@ def test_NN(self):
             b = torch.from_numpy(b.data)
             c = torch.from_numpy(c.data)
             x = torch.from_numpy(x.data)
-            y = torch.from_numpy(y.data).long()
+            y = torch.from_numpy(y.data)
         A.requires_grad = True
         B.requires_grad = True
         C.requires_grad = True