flaky test

Lightning-AI · Oct 11, 2023 · 5cdb642 · 5cdb642
1 parent 6e07148
commit 5cdb642
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/tests_pytorch/trainer/optimization/test_manual_optimization.py b/tests/tests_pytorch/trainer/optimization/test_manual_optimization.py
@@ -388,6 +388,7 @@ def on_before_optimizer_step(self, optimizer, *_):
 
 def test_step_with_optimizer_closure(tmpdir):
     """Tests that `step` works with optimizer_closure."""
+    seed_everything(1)
 
     class TestModel(BoringModel):
         _losses = []