fix test

pyt-team · Jul 13, 2024 · dd3ae92 · dd3ae92
1 parent b8f423d
commit dd3ae92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/transforms/liftings/pointcloud2simplicial/test_raindrop_drop_top_lifting.py b/test/transforms/liftings/pointcloud2simplicial/test_raindrop_drop_top_lifting.py
@@ -25,7 +25,7 @@ def setup_method(self):
             x=torch.randn(215, batch_size, 4),
             times=torch.randn(215, batch_size, 1),
             static=torch.randn(batch_size, 9),
-            y=torch.randint(0, 2, (batch_size)),
+            y=torch.randint(2, (batch_size,)),
         )
 
     def test_lift_topology(self):