Checkpoints.

This commit is contained in:
Robin Huang 2024-09-09 17:27:33 +09:00
parent b0a75f3f06
commit 54d7848d94

View File

@ -11,7 +11,7 @@ def mock_yaml_content():
return { return {
'test_config': { 'test_config': {
'base_path': '~/App/', '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) load_extra_path_config(dummy_yaml_file_name)
expected_calls = [ 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) assert mock_add_model_folder_path.call_count == len(expected_calls)