From 3b1ff4a7f43ec32716323b9aa7754cf85149c0ae Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sun, 10 Dec 2023 12:59:00 -0500 Subject: [PATCH] resolve test failure caused by renamed sqlite_database module --- invokeai/app/services/model_install/model_install_base.py | 3 ++- tests/app/services/model_install/test_model_install.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/invokeai/app/services/model_install/model_install_base.py b/invokeai/app/services/model_install/model_install_base.py index c2c11341df..81feaf85e7 100644 --- a/invokeai/app/services/model_install/model_install_base.py +++ b/invokeai/app/services/model_install/model_install_base.py @@ -6,7 +6,7 @@ from pathlib import Path from typing import Any, Dict, List, Literal, Optional, Union from fastapi import Body -from pydantic import BaseModel, Field, TypeAdapter, field_validator +from pydantic import BaseModel, Field, field_validator from pydantic.networks import AnyHttpUrl from typing_extensions import Annotated @@ -124,6 +124,7 @@ class URLModelSource(StringLikeSource): ModelSource = Annotated[Union[LocalModelSource, HFModelSource, URLModelSource], Body(discriminator="type")] + class ModelInstallJob(BaseModel): """Object that tracks the current status of an install request.""" diff --git a/tests/app/services/model_install/test_model_install.py b/tests/app/services/model_install/test_model_install.py index 607337fe85..eec6e033e5 100644 --- a/tests/app/services/model_install/test_model_install.py +++ b/tests/app/services/model_install/test_model_install.py @@ -18,7 +18,7 @@ from invokeai.app.services.model_install import ( ModelInstallServiceBase, ) from invokeai.app.services.model_records import ModelRecordServiceBase, ModelRecordServiceSQL, UnknownModelException -from invokeai.app.services.shared.sqlite import SqliteDatabase +from invokeai.app.services.shared.sqlite.sqlite_database import SqliteDatabase from invokeai.backend.model_manager.config import BaseModelType, ModelType from invokeai.backend.util.logging import InvokeAILogger