Allow no graphviz in .transport.build.get_computer()

iiasa · Dec 4, 2024 · 2cc9159 · 2cc9159
1 parent dc6d805
commit 2cc9159
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/message_ix_models/model/transport/build.py b/message_ix_models/model/transport/build.py
@@ -13,6 +13,7 @@
 from message_ix_models.model import bare, build
 from message_ix_models.util import minimum_version
 from message_ix_models.util._logging import mark_time
+from message_ix_models.util.graphviz import HAS_GRAPHVIZ
 
 from . import Config
 from .structure import get_technology_groups
@@ -460,7 +461,7 @@ def get_computer(
     # Add tasks for debugging the build
     add_debug(c)
 
-    if visualize:
+    if visualize and HAS_GRAPHVIZ:
         path = context.get_local_path("transport", "build.svg")
         path.parent.mkdir(exist_ok=True)
         c.visualize(filename=path, key="add transport data")