From 002f5b74a5f4e9c1df7a45df45e6be84d0c3eb05 Mon Sep 17 00:00:00 2001 From: Dylan <52908667+smellycloud@users.noreply.github.com> Date: Tue, 3 Dec 2024 10:59:25 +0100 Subject: [PATCH] fix test --- tests/test_model_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_model_manager.py b/tests/test_model_manager.py index 97e56b8..4a3ae86 100644 --- a/tests/test_model_manager.py +++ b/tests/test_model_manager.py @@ -230,7 +230,7 @@ def test_save_model_outputs(mock_model_path): df_output = pd.DataFrame({"col1": [5, 6], "col2": [7, 8]}) path_generated = Path("/path/to/generated") with patch("builtins.open", new_callable=mock_open), patch("pathlib.Path.mkdir"): - manager._save_model_outputs(df_evaluation, df_output, path_generated) + manager._save_model_outputs(df_evaluation, df_output, path_generated, sequence_number=1) with patch("pathlib.Path.exists", return_value=True): assert Path(f"{path_generated}/output_1_test_run_20220101.pkl").exists() assert Path(f"{path_generated}/evaluation_1_test_run_20220101.pkl").exists()