Visualizations as SVG with links and tooltips in jupyter notebook >=7.

.github/workflows/build.yml

@@ -357,7 +357,7 @@ jobs:
     - name: Fix tutorial dependencies
       if: ${{ matrix.dir == 'examples/kdd22' }}
       run: |
-        pip install "pandas<2.0.0" "xgboost<=1.5.1"
+        pip install "numpy<2.0.0" "pandas<2.0.0" "xgboost<=1.5.1"
     - name: pip list packages
       run: pip list
     - name: show pip dependencies

lale/helpers.py

@@ -394,7 +394,6 @@ def to_graphviz(
     lale_operator: "lale.operators.Operator",
     ipython_display: bool = True,
     call_depth: int = 1,
-    **dot_graph_attr,
 ):
     import lale.json_operator
     import lale.operators
@@ -403,7 +402,7 @@ def to_graphviz(
     if not isinstance(lale_operator, lale.operators.Operator):
         raise TypeError("The input to to_graphviz needs to be a valid LALE operator.")
     jsn = lale.json_operator.to_json(lale_operator, call_depth=call_depth + 1)
-    dot = lale.visualize.json_to_graphviz(jsn, ipython_display, dot_graph_attr)
+    dot = lale.visualize.json_to_graphviz(jsn, ipython_display)
     return dot
 
 

lale/visualize.py

@@ -125,7 +125,7 @@ def _indiv_op_tooltip(uid, jsn) -> str:
     return tooltip
 
 
-def _json_to_graphviz_rec(uid, jsn, cluster2reps, is_root, dot_graph_attr):
+def _json_to_graphviz_rec(uid, jsn, cluster2reps, is_root):
     kind = lale.json_operator.json_op_kind(jsn)
     dot: graphviz.Digraph
     if kind in ["Pipeline", "OperatorChoice"] or "steps" in jsn:
@@ -135,7 +135,9 @@ def _json_to_graphviz_rec(uid, jsn, cluster2reps, is_root, dot_graph_attr):
     if is_root:
         dot.attr(
             "graph",
-            {**dot_graph_attr, "rankdir": "LR", "compound": "true", "nodesep": "0.1"},
+            rankdir="LR",
+            compound="true",
+            nodesep="0.1",
         )
         dot.attr("node", fontsize="11", margin="0.06,0.03")
     if kind == "Pipeline":
@@ -183,7 +185,7 @@ def _json_to_graphviz_rec(uid, jsn, cluster2reps, is_root, dot_graph_attr):
     for step_uid, step_jsn in nodes.items():
         node_kind = lale.json_operator.json_op_kind(step_jsn)
         if node_kind in ["Pipeline", "OperatorChoice"] or "steps" in step_jsn:
-            sub_dot = _json_to_graphviz_rec(step_uid, step_jsn, cluster2reps, False, {})
+            sub_dot = _json_to_graphviz_rec(step_uid, step_jsn, cluster2reps, False)
             dot.subgraph(sub_dot)
         else:
             assert node_kind == "IndividualOp"
@@ -226,12 +228,22 @@ def _json_to_graphviz_rec(uid, jsn, cluster2reps, is_root, dot_graph_attr):
     return dot
 
 
-def json_to_graphviz(jsn, ipython_display, dot_graph_attr):
+class _HTML4Jupyter:
+    def __init__(self, html):
+        self.html = html
+
+    def _repr_html_(self):
+        return self.html
+
+
+def json_to_graphviz(jsn, ipython_display):
     cluster2reps = _get_cluster2reps(jsn)
-    dot = _json_to_graphviz_rec("(root)", jsn, cluster2reps, True, dot_graph_attr)
+    dot = _json_to_graphviz_rec("(root)", jsn, cluster2reps, True)
     if ipython_display:
         import IPython.display
 
-        IPython.display.display(dot)
+        svg = dot.pipe(format="svg", encoding="utf-8")
+        for_jupyter = _HTML4Jupyter(svg)
+        IPython.display.display(for_jupyter)
         return None
     return dot