From 5cdb642e6000fef8a620e7ba290b71d3928a3956 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Thu, 12 Oct 2023 00:33:47 +0200 Subject: [PATCH] flaky test --- .../trainer/optimization/test_manual_optimization.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/tests_pytorch/trainer/optimization/test_manual_optimization.py b/tests/tests_pytorch/trainer/optimization/test_manual_optimization.py index 12348ac707e9f..8476b07bf2b8e 100644 --- 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 = []