Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Skip hashing in test_heuristic_import_with_type

Browse Source
This commit is contained in:
Brandon Rising 2024-03-18 11:53:49 -04:00 committed by Brandon
parent 06abea8db0
commit 77a70a8a9c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/app/services/model_install/test_model_install.py
View File

@ -283,17 +283,19 @@ def test_404_download(mm2_installer: ModelInstallServiceBase, mm2_app_config: In
}, },
], ],
) )
@pytest.mark.timeout(timeout=20, method="thread") @pytest.mark.timeout(timeout=40, method="thread")
def test_heuristic_import_with_type(mm2_installer: ModelInstallServiceBase, model_params: Dict[str, str]): def test_heuristic_import_with_type(mm2_installer: ModelInstallServiceBase, model_params: Dict[str, str]):
"""Test whether or not type is respected on configs when passed to heuristic import.""" """Test whether or not type is respected on configs when passed to heuristic import."""
assert "name" in model_params and "type" in model_params assert "name" in model_params and "type" in model_params
config1: Dict[str, Any] = { config1: Dict[str, Any] = {
"name": f"{model_params['name']}_1", "name": f"{model_params['name']}_1",
"type": model_params["type"], "type": model_params["type"],
"hash": "placeholder1",
} }
config2: Dict[str, Any] = { config2: Dict[str, Any] = {
"name": f"{model_params['name']}_2", "name": f"{model_params['name']}_2",
"type": ModelType(model_params["type"]), "type": ModelType(model_params["type"]),
"hash": "placeholder2",
} }
assert "repo_id" in model_params assert "repo_id" in model_params
install_job1 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config1) install_job1 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config1)