Skip to content

Commit 6bf6966

Browse files
tests(mm): exclude UnknownModelConfig from model config tests
1 parent 15e5c9a commit 6bf6966

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/test_model_probe.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
MainDiffusersConfig,
1818
ModelConfigBase,
1919
ModelConfigFactory,
20+
UnknownModelConfig,
2021
get_model_discriminator_value,
2122
)
2223
from invokeai.backend.model_manager.legacy_probe import (
@@ -257,6 +258,10 @@ def test_any_model_config_includes_all_config_classes():
257258
extracted.add(config_class)
258259

259260
expected = set(ModelConfigBase.all_config_classes()) - {MinimalConfigExample}
261+
262+
# UnknownModelConfig is deliberately excluded from AnyModelConfig to avoid it being used for model probing.
263+
expected.remove(UnknownModelConfig)
264+
260265
assert extracted == expected
261266

262267

0 commit comments

Comments
 (0)