Use display names in visualization

gyorilab · Jul 26, 2023 · d240866 · d240866
1 parent 3fafe28
commit d240866
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mira/modeling/viz.py b/mira/modeling/viz.py
@@ -25,7 +25,10 @@ def __init__(self, model: Model):
         for variable in model.variables.values():
             identifiers = variable.data.get('identifiers')
             contexts = variable.data.get('context')
-            name = variable.data.get('name', str(variable.key))
+            if variable.concept.display_name:
+                name = variable.concept.display_name
+            else:
+                name = variable.data.get('name', str(variable.key))
             if not identifiers and not contexts:
                 label = name
                 shape = "oval"