This commit is contained in:
Brandon Rising 2024-03-18 10:11:56 -04:00
parent 9b6f3bded9
commit 62d7e38030

View File

@ -5,7 +5,6 @@ Test the model installer
import platform import platform
import uuid import uuid
from pathlib import Path from pathlib import Path
from time import sleep
from typing import Any, Dict from typing import Any, Dict
import pytest import pytest
@ -300,7 +299,7 @@ def test_heuristic_import_with_type(mm2_installer: ModelInstallServiceBase, mode
mm2_installer.wait_for_installs(timeout=10) mm2_installer.wait_for_installs(timeout=10)
if model_params["type"] != "embedding": if model_params["type"] != "embedding":
assert install_job1.errored assert install_job1.errored
assert install_job1.error_type == 'InvalidModelConfigException' assert install_job1.error_type == "InvalidModelConfigException"
return return
assert install_job1.complete assert install_job1.complete
assert install_job1.config_out if model_params["type"] == "embedding" else not install_job1.config_out assert install_job1.config_out if model_params["type"] == "embedding" else not install_job1.config_out