From 094d4eeddc6a71e80bb61d620dd2c543b3f79a94 Mon Sep 17 00:00:00 2001
From: Tom White <tom.e.white@gmail.com>
Date: Thu, 8 Jan 2026 14:33:55 +0000
Subject: [PATCH] Fix to allow cubed to work without diagnostics

---
 cubed/core/plan.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cubed/core/plan.py b/cubed/core/plan.py
index 3965a46fa..03370f862 100644
--- a/cubed/core/plan.py
+++ b/cubed/core/plan.py
@@ -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
@@ -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,