refactor(quality): use the precise assert in the model status test

This commit is contained in:
Vito Sansevero
2026-10-30 21:53:52 -07:00
parent 14cffa86f2
commit 4b89162c76
+3 -1
View File
@@ -1178,7 +1178,9 @@ class TestModelStatus(AutotagAPITestCase):
)
# Kept fields legacy pages and the workspace read.
for model in models.values():
self.assertTrue({"name", "description", "size", "downloaded"} <= set(model))
self.assertLessEqual(
{"name", "description", "size", "downloaded"}, set(model)
)
self.assertTrue(data["model_loaded"])
self.assertEqual(data["loaded_model_type"], "wd14-vit")