Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion cubed/core/plan.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
import networkx as nx

from cubed.core.optimization import is_input_array, multiple_inputs_optimize_dag
from cubed.diagnostics.colors import APRICOT, LAVENDER, RED
from cubed.primitive.blockwise import BlockwiseSpec
from cubed.primitive.types import PrimitiveOperation
from cubed.runtime.pipeline import visit_node_generations
Expand Down Expand Up @@ -612,6 +611,8 @@ def visualize(
show_hidden=False,
engine: Literal["cytoscape", "graphviz"] | None = None,
):
from cubed.diagnostics.colors import APRICOT, LAVENDER, RED

if engine == "cytoscape":
return self.visualize_cytoscape(
filename,
Expand Down
Loading