icrdr/AnyImage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
马南
2026-09-01 12:30:48 +08:00
parent c97b4a3e41
commit 4d4b892d64
2 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/anyimage/properties.py
View File

@@ -161,7 +161,7 @@ def _model_items(models):
return [ return [
( (
model["key"], model["key"],
model["label"] if model.get("ready") else f"{model['label']} (Missing)", model["label"],
f"{model['description']}; license: {model['license']}", f"{model['description']}; license: {model['license']}",
index, index,
) )

9
tests/test_blender_addon.py
View File

@@ -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 depth_items[:-1]), expected)
self.assertEqual(tuple(item[0] for item in mesh_depth_items), 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): def test_required_model_operator_requests_both_models(self):
operator = self.anyimage.operators.DownloadRequiredModels() operator = self.anyimage.operators.DownloadRequiredModels()