From 54d7848d946d8a1db841527350f8dc6324cf9595 Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Mon, 9 Sep 2024 17:27:33 +0900 Subject: [PATCH] Checkpoints. --- tests-unit/main_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests-unit/main_test.py b/tests-unit/main_test.py index 040c1a7cf..836c94995 100644 --- a/tests-unit/main_test.py +++ b/tests-unit/main_test.py @@ -11,7 +11,7 @@ def mock_yaml_content(): return { 'test_config': { 'base_path': '~/App/', - 'model1': 'subfolder1', + 'checkpoints': 'subfolder1', } } @@ -53,7 +53,7 @@ def test_load_extra_model_paths_expands_userpath( load_extra_path_config(dummy_yaml_file_name) expected_calls = [ - ('model1', os.path.join(mock_expanded_home, 'App', 'subfolder1')), + ('checkpoints', os.path.join(mock_expanded_home, 'App', 'subfolder1')), ] assert mock_add_model_folder_path.call_count == len(expected_calls)