work on graph

ecmwf · Jun 15, 2024 · 43a6e22 · 43a6e22
1 parent 6ede607
commit 43a6e22
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/anemoi/inference/checkpoint/metadata/__init__.py b/src/anemoi/inference/checkpoint/metadata/__init__.py
@@ -350,8 +350,13 @@ def digraph(self, label_maker=lambda x: dict(label=x.kind)):
 
         for node, label in nodes.items():
             for k, v in label.items():
-                # Use json.dumps to escape special characters
-                label[k] = json.dumps(v)
+
+                if isinstance(v, str) and v.startswith("<") and v.endswith(">"):
+                    # Skip HTML labels
+                    pass
+                else:
+                    # Use json.dumps to escape special characters
+                    label[k] = json.dumps(v)
 
             label = " ".join([f"{k}={v}" for k, v in label.items()])