From 7cdc821801663ac9355a686fb2d30510c1ce714d Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Thu, 18 Sep 2025 21:56:04 +1000 Subject: [PATCH] tests(mm): fix test for MM, leave the UnknownModelConfig class in the list of configs --- invokeai/backend/model_manager/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/backend/model_manager/config.py b/invokeai/backend/model_manager/config.py index b0b46b75ec..83f0c1d2bf 100644 --- a/invokeai/backend/model_manager/config.py +++ b/invokeai/backend/model_manager/config.py @@ -175,7 +175,7 @@ class ModelConfigBase(ABC, BaseModel): if issubclass(cls, LegacyProbeMixin): ModelConfigBase.USING_LEGACY_PROBE.add(cls) # Cannot use `elif isinstance(cls, UnknownModelConfig)` because UnknownModelConfig is not defined yet - elif cls.__name__ != "UnknownModelConfig": + else: ModelConfigBase.USING_CLASSIFY_API.add(cls) @staticmethod