From 1868483b0d9b67446f1bc4e22f76bf20b83322aa Mon Sep 17 00:00:00 2001 From: Nicholas Garcia Date: Mon, 29 Jan 2024 11:08:50 -0800 Subject: [PATCH] Fix MLFlowLogger test (#2912) Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com> --- tests/loggers/test_mlflow_logger.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/loggers/test_mlflow_logger.py b/tests/loggers/test_mlflow_logger.py index 039c4a33951..102d6b9d584 100644 --- 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(