From a4a31fe3649113facc668d5a9664ec9f57869bf8 Mon Sep 17 00:00:00 2001
From: s214735 <s214735@dtu.dk>
Date: Mon, 30 Dec 2024 10:19:48 +0100
Subject: [PATCH] fix

---
 qim3d/utils/_doi.py | 2 +-
 qim3d/viz/_cc.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/qim3d/utils/_doi.py b/qim3d/utils/_doi.py
index 4c3ec5f5..36302e97 100644
--- a/qim3d/utils/_doi.py
+++ b/qim3d/utils/_doi.py
@@ -56,7 +56,7 @@ def get_bibtex(doi):
 
     return _log_and_get_text(doi, header)
 
-def custom_header(doi, header):
+def custom_header(doi: str, header: str) -> str:
     """Allows a custom header to be passed
 
     Example:
diff --git a/qim3d/viz/_cc.py b/qim3d/viz/_cc.py
index 0bd03466..7b6cb212 100644
--- a/qim3d/viz/_cc.py
+++ b/qim3d/viz/_cc.py
@@ -23,7 +23,7 @@ def plot_cc(
         connected_components (CC): The connected components object.
         component_indexs (list | tuple, optional): The components to plot. If None the first max_cc_to_plot=32 components will be plotted. Defaults to None.
         max_cc_to_plot (int, optional): The maximum number of connected components to plot. Defaults to 32.
-        overlay (optional): Overlay image. Defaults to None.
+        overlay (np.ndarray, optional): Overlay image. Defaults to None.
         crop (bool, optional): Whether to crop the image to the cc. Defaults to False.
         display_figure (bool, optional): Whether to show the figure. Defaults to True.
         color_map (str, optional): Specifies the color map for the image. Defaults to "viridis".
-- 
GitLab