Add workaround for pandas 2.0

GenericMappingTools · Dec 12, 2024 · 068c5cb · 068c5cb
1 parent 5867999
commit 068c5cb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pygmt/clib/conversion.py b/pygmt/clib/conversion.py
@@ -195,7 +195,12 @@ def _to_numpy(data: Any) -> np.ndarray:
 
     # Deal with timezone-aware datetime dtypes.
     if getattr(dtype, "tz", None):  # pandas.DatetimeTZDtype
-        numpy_dtype = getattr(dtype, "base", None)
+        if pd.__version__ < Version("2.1"):
+            # Workaround for bug https://github.com/pandas-dev/pandas/issues/52705.
+            # Solution from https://github.com/pandas-dev/pandas/pull/52706.
+            numpy_dtype = np.dtype(f"M8[{dtype.unit}]")
+        else:
+            numpy_dtype = getattr(dtype, "base", None)
     elif getattr(getattr(dtype, "pyarrow_dtype", None), "tz", None):
         # pd.ArrayDtype[pa.Timestamp]
         numpy_dtype = getattr(dtype, "numpy_dtype", None)