TYP: overload tz_to_dtype (#56260)

pandas-dev · Nov 30, 2023 · adcb36e · adcb36e
1 parent 6093074
commit adcb36e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pandas/core/arrays/datetimes.py b/pandas/core/arrays/datetimes.py
@@ -8,6 +8,7 @@
 from typing import (
     TYPE_CHECKING,
     cast,
+    overload,
 )
 import warnings
 
@@ -93,6 +94,16 @@
 _ITER_CHUNKSIZE = 10_000
 
 
+@overload
+def tz_to_dtype(tz: tzinfo, unit: str = ...) -> DatetimeTZDtype:
+    ...
+
+
+@overload
+def tz_to_dtype(tz: None, unit: str = ...) -> np.dtype[np.datetime64]:
+    ...
+
+
 def tz_to_dtype(
     tz: tzinfo | None, unit: str = "ns"
 ) -> np.dtype[np.datetime64] | DatetimeTZDtype: