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
parent 97bcd54408
commit c778d74a42
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)