From c6b6491f2435597c9307bc49daa27f10fb4d8690 Mon Sep 17 00:00:00 2001
From: Akhilesh Halageri <akhilesh@zetta.ai>
Date: Tue, 12 Sep 2023 20:12:06 +0000
Subject: [PATCH] fix: circular import

---
 pychunkedgraph/debug/utils.py | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/pychunkedgraph/debug/utils.py b/pychunkedgraph/debug/utils.py
index e194f4ee1..53152ec6f 100644
--- a/pychunkedgraph/debug/utils.py
+++ b/pychunkedgraph/debug/utils.py
@@ -2,9 +2,6 @@
 
 import numpy as np
 
-from ..graph import ChunkedGraph
-from ..graph.utils.basetypes import NODE_ID
-
 
 def print_attrs(d):
     for k, v in d.items():
@@ -18,12 +15,7 @@ def print_attrs(d):
             print(v)
 
 
-def print_node(
-    cg: ChunkedGraph,
-    node: NODE_ID,
-    indent: int = 0,
-    stop_layer: int = 2,
-) -> None:
+def print_node(cg, node: np.uint64, indent: int = 0, stop_layer: int = 2) -> None:
     children = cg.get_children(node)
     print(f"{' ' * indent}{node}[{len(children)}]")
     if cg.get_chunk_layer(node) <= stop_layer:
@@ -32,8 +24,8 @@ def print_node(
         print_node(cg, child, indent=indent + 4, stop_layer=stop_layer)
 
 
-def get_l2children(cg: ChunkedGraph, node: NODE_ID) -> np.ndarray:
-    nodes = np.array([node], dtype=NODE_ID)
+def get_l2children(cg, node: np.uint64) -> np.ndarray:
+    nodes = np.array([node], dtype=np.uint64)
     layers = cg.get_chunk_layers(nodes)
     assert np.all(layers > 2), "nodes must be at layers > 2"
     l2children = []