Correct types in '_check_attributes' docstring

mllam · Dec 13, 2024 · e907a6d · e907a6d
1 parent e3c0f22
commit e907a6d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllam_data_prep/derived_variables.py b/mllam_data_prep/derived_variables.py
@@ -212,15 +212,15 @@ def _check_attributes(field, field_attributes):
 
     Parameters
     ----------
-    field: Union[xr.DataArray, Tuple[xr.DataArray]]
+    field: xr.DataArray
         The derived field
     field_attributes: Dict[str, str]
         Dictionary with attributes for the derived variables.
         Defined in the config file.
 
     Returns
     -------
-    field: Union[xr.DataArray, Tuple[xr.DataArray]]
+    field: xr.DataArray
         The derived field
     """
     for attribute in ["units", "long_name"]: