REF: move SparseDtype, ArrowDtype to dtypes.dtypes (#53116)

* REF: move SparseDtype to dtypes.dtypes * REF: move ArrowDtype to dtypes.dtypes
pandas-dev · May 15, 2023 · 2b107bf · 2b107bf
1 parent 4f88a2a
commit 2b107bf
Show file tree

Hide file tree

Showing 37 changed files with 841 additions and 875 deletions.
diff --git a/pandas/__init__.py b/pandas/__init__.py
@@ -108,7 +108,7 @@
     DataFrame,
 )
 
-from pandas.core.arrays.sparse import SparseDtype
+from pandas.core.dtypes.dtypes import SparseDtype
 
 from pandas.tseries.api import infer_freq
 from pandas.tseries import offsets

diff --git a/pandas/compat/pickle_compat.py b/pandas/compat/pickle_compat.py
@@ -134,6 +134,10 @@ def load_reduce(self):
         "pandas.core.indexes.base",
         "Index",
     ),
+    ("pandas.core.arrays.sparse.dtype", "SparseDtype"): (
+        "pandas.core.dtypes.dtypes",
+        "SparseDtype",
+    ),
 }
 
 

diff --git a/pandas/core/api.py b/pandas/core/api.py
@@ -7,6 +7,7 @@
 from pandas._libs.missing import NA
 
 from pandas.core.dtypes.dtypes import (
+    ArrowDtype,
     CategoricalDtype,
     DatetimeTZDtype,
     IntervalDtype,
@@ -25,7 +26,6 @@
     value_counts,
 )
 from pandas.core.arrays import Categorical
-from pandas.core.arrays.arrow import ArrowDtype
 from pandas.core.arrays.boolean import BooleanDtype
 from pandas.core.arrays.floating import (
     Float32Dtype,

diff --git a/pandas/core/arrays/arrow/__init__.py b/pandas/core/arrays/arrow/__init__.py
@@ -1,4 +1,3 @@
 from pandas.core.arrays.arrow.array import ArrowExtensionArray
-from pandas.core.arrays.arrow.dtype import ArrowDtype
 
-__all__ = ["ArrowDtype", "ArrowExtensionArray"]
+__all__ = ["ArrowExtensionArray"]
diff --git a/pandas/core/arrays/arrow/array.py b/pandas/core/arrays/arrow/array.py
@@ -60,8 +60,9 @@
     import pyarrow as pa
     import pyarrow.compute as pc
 
+    from pandas.core.dtypes.dtypes import ArrowDtype
+
     from pandas.core.arrays.arrow._arrow_utils import fallback_performancewarning
-    from pandas.core.arrays.arrow.dtype import ArrowDtype
 
     ARROW_CMP_FUNCS = {
         "eq": pc.equal,

diff --git a/pandas/core/arrays/arrow/dtype.py b/pandas/core/arrays/arrow/dtype.py
diff --git a/pandas/core/arrays/sparse/__init__.py b/pandas/core/arrays/sparse/__init__.py
@@ -8,14 +8,12 @@
     SparseArray,
     make_sparse_index,
 )
-from pandas.core.arrays.sparse.dtype import SparseDtype
 
 __all__ = [
     "BlockIndex",
     "IntIndex",
     "make_sparse_index",
     "SparseAccessor",
     "SparseArray",
-    "SparseDtype",
     "SparseFrameAccessor",
 ]