diff --git a/test/transforms/test_random_jitter.py b/test/transforms/test_random_jitter.py index 028cff2f4b15..88e81cb3d08a 100644 --- 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