TST Fix test collection (#792)

A cuda-related test was failing but since cuda is not run by CI, it wasn't caught.
skorch-dev · Jul 11, 2021 · 359c15b · 359c15b
1 parent 906932c
commit 359c15b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skorch/tests/test_utils.py b/skorch/tests/test_utils.py
@@ -100,8 +100,8 @@ def parameters():
                 ),
                 (torch.as_tensor(55), torch.as_tensor(55).to('cuda')),
                 (
-                    pack_padded_sequence(x, y),
-                    pack_padded_sequence(x, y).to('cuda')
+                    pack_padded_sequence(x, y.to('cpu')),
+                    pack_padded_sequence(x, y.to('cpu')).to('cuda')
                 ),
         ]:
             yield X, expected, device