fix: Update model label handling to always display label without missing status

2026-09-01 12:30:48 +08:00
parent c97b4a3e41
commit 4d4b892d64
2 changed files with 10 additions and 1 deletions
--- a/tests/test_blender_addon.py
+++ b/tests/test_blender_addon.py
@@ -559,6 +559,15 @@ class BlenderAddonTest(unittest.TestCase):
        )
        self.assertEqual(tuple(item[0] for item in depth_items[:-1]), expected)
        self.assertEqual(tuple(item[0] for item in mesh_depth_items), expected)
+        expected_labels = (
+            "MoGe-2 ViT-S",
+            "MoGe-2 ViT-B",
+            "MoGe-2 ViT-L",
+        )
+        self.assertEqual(
+            tuple(item[1] for item in depth_items[:-1]), expected_labels
+        )
+        self.assertEqual(tuple(item[1] for item in mesh_depth_items), expected_labels)

    def test_required_model_operator_requests_both_models(self):
        operator = self.anyimage.operators.DownloadRequiredModels()