docs: gather dims union types

nci · Aug 16, 2023 · 020d857 · 020d857
1 parent 0518923
commit 020d857
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/scores/utils.py b/src/scores/utils.py
@@ -44,8 +44,8 @@ def gather_dimensions(fcst_dims, obs_dims, weights_dims=None, reduce_dims=None,
         fcst_dims (Iterable[str]): Forecast dimensions inputs
         obs_dims (Iterable[str]): Observation dimensions inputs.
         weights_dims (Iterable[str]): Weight dimension inputs.
-        reduce_dims (Iterable[str]): Dimensions to reduce.
-        preserve_dims (Iterable[str]): Dimensions to preserve.
+        reduce_dims (Union[str, Iterable[str]]): Dimensions to reduce.
+        preserve_dims (Union[str, Iterable[str]]): Dimensions to preserve.
 
     Returns:
         Tuple[str]: Dimensions based on optional args.