Improve type hints

GenericMappingTools · Mar 29, 2024 · d5283cd · d5283cd
1 parent 065ec12
commit d5283cd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pygmt/datatypes/dataset.py b/pygmt/datatypes/dataset.py
@@ -3,7 +3,8 @@
 """
 
 import ctypes as ctp
-from typing import ClassVar
+from collections.abc import Mapping
+from typing import Any, ClassVar
 
 import numpy as np
 import pandas as pd
@@ -145,8 +146,8 @@ class _GMT_DATASEGMENT(ctp.Structure):  # noqa: N801
 
     def to_dataframe(
         self,
-        column_names: list[str] | None = None,
-        dtype: type | dict[str, type] | None = None,
+        column_names: pd.Index[Any] | None = None,
+        dtype: type | Mapping[Any, type] | None = None,
         index_col: str | int | None = None,
     ) -> pd.DataFrame:
         """