Improve the checking of dtypes

GenericMappingTools · Dec 12, 2024 · 18057c1 · 18057c1
1 parent c188779
commit 18057c1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pygmt/clib/conversion.py b/pygmt/clib/conversion.py
@@ -194,10 +194,10 @@ def _to_numpy(data: Any) -> np.ndarray:
             data = data.to_numpy(na_value=np.nan)
 
     # Deal with timezone-aware datetime dtypes.
-    if getattr(dtype, "tz", None):  # pandas.DatetimeTZDtype
+    if isinstance(dtype, pd.DatetimeTZDtype):  # pandas.DatetimeTZDtype
         numpy_dtype = getattr(dtype, "base", None)
-    elif getattr(dtype, "pyarrow_dtype", None) and hasattr(dtype.pyarrow_dtype, "tz"):
-        # pd.ArrayDtype[pa.Timestamp]
+    elif isinstance(dtype, pd.ArrowDtype) and hasattr(dtype.pyarrow_dtype, "tz"):
+        # pd.ArrowDtype[pa.Timestamp]
         numpy_dtype = getattr(dtype, "numpy_dtype", None)
         if Version(pd.__version__) < Version("2.1"):
             # In pandas 2.0, dtype.numpy_type is dtype("O").