Fix MLFlowLogger test (mosaicml#2912)

Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com>
ShashankMosaicML · Feb 3, 2024 · 1868483 · 1868483
1 parent 3fc2efe
commit 1868483
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/loggers/test_mlflow_logger.py b/tests/loggers/test_mlflow_logger.py
@@ -646,6 +646,8 @@ def test_mlflow_ignore_hyperparameters(tmp_path):
                                       ignore_hyperparameters=['num*', 'mlflow_run_id', 'nothing'])
 
     Trainer(model=SimpleConvModel(), loggers=test_mlflow_logger, max_duration=f'4ba')
+    # Allow async logging to finish.
+    time.sleep(3)
     test_mlflow_logger.post_close()
 
     run = _get_latest_mlflow_run(