From cb61b60c2792e0caadfa7d8c2868c155e1118cbf Mon Sep 17 00:00:00 2001
From: Vito Sansevero <vito.sansevero@gmail.com>
Date: Mon, 4 Aug 2025 14:40:02 -0700
Subject: [PATCH] fix: update imports to use new simplified XYZPlotController

- Replace XYZPlotControllerAdvanced with XYZPlotController
- Update all import statements to match new class name
- Fix __all__ exports in xyz_grid module
---
 kikotools/__init__.py                | 4 ++--
 kikotools/tools/xyz_grid/__init__.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/kikotools/__init__.py b/kikotools/__init__.py
index 785e4d8..08341e8 100644
--- a/kikotools/__init__.py
+++ b/kikotools/__init__.py
@@ -14,7 +14,7 @@ from .tools.image_scale_down_by import ImageScaleDownByNode
 from .tools.gemini_prompt import GeminiPromptNode
 from .tools.display_any import DisplayAnyNode
 from .tools.display_text import DisplayTextNode
-from .tools.xyz_grid import XYZPlotControllerAdvanced, ImageGridCombiner
+from .tools.xyz_grid import XYZPlotController, ImageGridCombiner
 
 # ComfyUI node registration mappings
 NODE_CLASS_MAPPINGS = {
@@ -30,7 +30,7 @@ NODE_CLASS_MAPPINGS = {
     "GeminiPrompt": GeminiPromptNode,
     "DisplayAny": DisplayAnyNode,
     "DisplayText": DisplayTextNode,
-    "XYZPlotController": XYZPlotControllerAdvanced,
+    "XYZPlotController": XYZPlotController,
     "ImageGridCombiner": ImageGridCombiner,
 }
 
diff --git a/kikotools/tools/xyz_grid/__init__.py b/kikotools/tools/xyz_grid/__init__.py
index 2a8807d..cc2c10b 100644
--- a/kikotools/tools/xyz_grid/__init__.py
+++ b/kikotools/tools/xyz_grid/__init__.py
@@ -13,4 +13,4 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "ImageGridCombiner": "Image Grid Combiner",
 }
 
-__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "XYZPlotController", "ImageGridCombiner"]
\ No newline at end of file
+__all__ = ["XYZPlotController", "ImageGridCombiner"]
\ No newline at end of file