diff --git a/qim3d/viz/_cc.py b/qim3d/viz/_cc.py index 7b6cb2127dbb1ddc5baad4a9d6d129744a800182..ebb67d036c152a379832dbbc066660fe08adab2b 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,