typo

pyg-team · Jun 20, 2022 · 4ea6807 · 4ea6807
1 parent 3e306f4
commit 4ea6807
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/transforms/test_random_jitter.py b/test/transforms/test_random_jitter.py
@@ -21,7 +21,7 @@ def test_random_jitter():
     assert data.pos.max().item() <= 0.1
 
     data = Data(pos=pos)
-    data = RandomJitter(0.1)(data)
+    data = RandomJitter([0.1, 1])(data)
     assert len(data) == 1
     assert data.pos[:, 0].min().item() >= -0.1
     assert data.pos[:, 0].max().item() <= 0.1