tests(mm): fix test for MM, leave the UnknownModelConfig class in the list of configs

This commit is contained in:
psychedelicious
2025-09-18 21:56:04 +10:00
parent 08853f9be2
commit 7cdc821801

View File

@@ -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