fix test for cuda conversion

lanpa · Sep 25, 2017 · 709b251 · 709b251
1 parent 8f8d6c9
commit 709b251
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_pytorch_np.py b/tests/test_pytorch_np.py
@@ -18,7 +18,7 @@ def test_pytorch_np():
 
         # CUDA variable
         if torch.cuda.device_count()>0:
-            assert isinstance(x2num.makenp(torch.autograd.variable.Variable(tensor)).cuda(), np.ndarray)
+            assert isinstance(x2num.makenp(torch.autograd.variable.Variable(tensor).cuda()), np.ndarray)
 
     # python primitive type
     assert(isinstance(x2num.makenp(0), np.ndarray))