Update times.py

pydata · Jul 9, 2024 · 0ec9be7 · 0ec9be7
1 parent 36831ce
commit 0ec9be7
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/xarray/coding/times.py b/xarray/coding/times.py
@@ -706,6 +706,20 @@ def _cast_to_dtype_if_safe(num: np.ndarray, dtype: np.dtype) -> np.ndarray:
     return cast_num
 
 
+@overload
+def encode_cf_datetime(
+    dates: chunkedduckarray,
+    units: str | None = None,
+    calendar: str | None = None,
+    dtype: np.dtype | None = None,
+) -> tuple[chunkedduckarray, str, str]: ...
+@overload
+def encode_cf_datetime(
+    dates: duckarray,
+    units: str | None = None,
+    calendar: str | None = None,
+    dtype: np.dtype | None = None,
+) -> tuple[duckarray, str, str]: ...
 def encode_cf_datetime(
     dates: duckarray | chunkedduckarray,
     units: str | None = None,