chore: update type annotation and docstring for closest_resolution()

juaml · Feb 9, 2024 · 1ffaeb4 · 1ffaeb4
1 parent fb58b90
commit 1ffaeb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/junifer/data/utils.py b/junifer/data/utils.py
@@ -8,14 +8,14 @@
 
 
 def closest_resolution(
-    resolution: Optional[float],
+    resolution: Optional[Union[float, int]],
     valid_resolution: Union[List[float], List[int], np.ndarray],
 ) -> Union[float, int]:
     """Find the closest resolution.
 
     Parameters
     ----------
-    resolution : float, optional
+    resolution : float or int, optional
         The given resolution. If None, will return the highest resolution
         (default None).
     valid_resolution : list of float or int, or np.ndarray