diff --git a/tests/test_time.py b/tests/test_time.py index af1666f692b..a93c352274e 100644 --- a/tests/test_time.py +++ b/tests/test_time.py @@ -141,7 +141,7 @@ def test_no_shuffle(self): """ Testing the staged activation without shuffling. """ - a = SomeModel() + a = SomeModel(shuffle=False) print(a) model = MockModel(shuffle=False) model.step()