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

Fix test to run on windows vms

Browse Source
This commit is contained in:
Brandon Rising 2024-03-18 10:10:45 -04:00 committed by Brandon
parent 9d5b96c119
commit 6968a068bb
1 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -295,18 +295,17 @@ def test_heuristic_import_with_type(mm2_installer: ModelInstallServiceBase, mode
"name": f"{model_params['name']}_2",
"type": ModelType(model_params["type"]),
}
try:
assert "repo_id" in model_params
install_job1 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config1)
assert "repo_id" in model_params
install_job1 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config1)
mm2_installer.wait_for_installs(timeout=10)
if model_params["type"] != "embedding":
assert install_job1.errored
assert install_job1.error_type == 'InvalidModelConfigException'
return
assert install_job1.complete
assert install_job1.config_out if model_params["type"] == "embedding" else not install_job1.config_out
while not install_job1.in_terminal_state:
sleep(0.01)
assert install_job1.config_out if model_params["type"] == "embedding" else not install_job1.config_out
install_job2 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config2)
while not install_job2.in_terminal_state:
sleep(0.01)
assert install_job2.config_out if model_params["type"] == "embedding" else not install_job2.config_out
except InvalidModelConfigException:
assert model_params["type"] != "embedding"
install_job2 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config2)
mm2_installer.wait_for_installs(timeout=10)
assert install_job2.complete
assert install_job2.config_out if model_params["type"] == "embedding" else not install_job2.config_out