From 4f8727e58dfa91d19698e8f0deb70a8b22817624 Mon Sep 17 00:00:00 2001
From: s214735 <s214735@dtu.dk>
Date: Mon, 30 Dec 2024 10:20:58 +0100
Subject: [PATCH] fix

---
 qim3d/viz/_cc.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/qim3d/viz/_cc.py b/qim3d/viz/_cc.py
index 7b6cb212..ebb67d03 100644
--- a/qim3d/viz/_cc.py
+++ b/qim3d/viz/_cc.py
@@ -2,15 +2,15 @@ import matplotlib.pyplot as plt
 import numpy as np
 import qim3d
 from qim3d.utils._logger import log
-
+from segmentation._connected_components import CC
 
 def plot_cc(
-    connected_components,
+    connected_components: CC,
     component_indexs: list | tuple = None,
-    max_cc_to_plot=32,
-    overlay=None,
-    crop=False,
-    display_figure=True,
+    max_cc_to_plot: int = 32,
+    overlay: np.ndarray = None,
+    crop: bool = False,
+    display_figure: bool = True,
     color_map: str = "viridis",
     value_min: float = None,
     value_max: float = None,
-- 
GitLab