diff --git a/src/anyimage/properties.py b/src/anyimage/properties.py index 0694a6e..3df199f 100644 --- a/src/anyimage/properties.py +++ b/src/anyimage/properties.py @@ -161,7 +161,7 @@ def _model_items(models): return [ ( model["key"], - model["label"] if model.get("ready") else f"{model['label']} (Missing)", + model["label"], f"{model['description']}; license: {model['license']}", index, ) diff --git a/tests/test_blender_addon.py b/tests/test_blender_addon.py index 2ac32fb..fb9f416 100644 --- 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()