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 {
'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)