try fix test with increased timeout

zauberzeug · Oct 24, 2024 · 0d98730 · 0d98730
1 parent b49ff45
commit 0d98730
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/learning_loop_node/tests/trainer/states/test_state_upload_model.py b/learning_loop_node/tests/trainer/states/test_state_upload_model.py
@@ -64,7 +64,7 @@ async def test_bad_server_response_content(test_initialized_trainer: TestingTrai
 
     await assert_training_state(trainer.training, TrainerState.TrainModelUploading, timeout=1, interval=0.001)
     # TODO goes to finished because of the error
-    await assert_training_state(trainer.training, TrainerState.ReadyForCleanup, timeout=2, interval=0.001)
+    await assert_training_state(trainer.training, TrainerState.ReadyForCleanup, timeout=5, interval=0.001)
 
     assert trainer_has_error(trainer)
     assert trainer.training.training_state == TrainerState.ReadyForCleanup