GH1227 align init pyi with init py from pandas (#1229)

* GH1227 Align __init__ file content with pandas

* GH1227 Align __init__.pyi with __init__.py

* GH1227 PR feedback

Author: loicdiridollou

pandas-stubs/_config/__init__.pyi

@@ -1,4 +1,4 @@
-from .config import (
+from pandas._config.config import (
     describe_option as describe_option,
     get_option as get_option,
     option_context as option_context,

pandas-stubs/api/extensions/__init__.pyi

@@ -11,7 +11,7 @@ from pandas.core.arrays import (
 
 from pandas._libs.lib import no_default as no_default
 
-from pandas.core.dtypes.dtypes import (
+from pandas.core.dtypes.base import (
     ExtensionDtype as ExtensionDtype,
     register_extension_dtype as register_extension_dtype,
 )

pandas-stubs/api/indexers/__init__.pyi

@@ -1,6 +1,6 @@
-from pandas.core.indexers import (
+from pandas.core.indexers import check_array_indexer as check_array_indexer
+from pandas.core.indexers.objects import (
     BaseIndexer as BaseIndexer,
     FixedForwardWindowIndexer as FixedForwardWindowIndexer,
     VariableOffsetWindowIndexer as VariableOffsetWindowIndexer,
-    check_array_indexer as check_array_indexer,
 )

pandas-stubs/core/dtypes/base.pyi

@@ -1,6 +1,7 @@
 from typing import (
     ClassVar,
     Literal,
+    TypeVar,
 )
 
 from pandas.core.arrays import ExtensionArray
@@ -28,3 +29,7 @@ class ExtensionDtype:
     def is_dtype(cls, dtype: object) -> bool: ...
 
 class StorageExtensionDtype(ExtensionDtype): ...
+
+_ExtensionDtypeT = TypeVar("_ExtensionDtypeT", bound=ExtensionDtype)
+
+def register_extension_dtype(cls: type[_ExtensionDtypeT]) -> type[_ExtensionDtypeT]: ...

pandas-stubs/core/dtypes/dtypes.pyi

@@ -2,7 +2,6 @@ import datetime as dt
 from typing import (
     Any,
     Literal,
-    TypeVar,
 )
 
 import numpy as np
@@ -21,11 +20,10 @@ from pandas._typing import (
     npt,
 )
 
-from .base import ExtensionDtype as ExtensionDtype
-
-_ExtensionDtypeT = TypeVar("_ExtensionDtypeT", bound=ExtensionDtype)
-
-def register_extension_dtype(cls: type[_ExtensionDtypeT]) -> type[_ExtensionDtypeT]: ...
+from pandas.core.dtypes.base import (
+    ExtensionDtype as ExtensionDtype,
+    register_extension_dtype as register_extension_dtype,
+)
 
 class BaseMaskedDtype(ExtensionDtype): ...
 class PandasExtensionDtype(ExtensionDtype): ...

pandas-stubs/core/interchange/__init__.pyi

@@ -0,0 +1,2 @@
+from pandas.core.interchange.dataframe_protocol import DataFrame as DataFrame
+from pandas.core.interchange.from_dataframe import from_dataframe as from_dataframe

pandas-stubs/io/__init__.pyi

@@ -0,0 +1,5 @@
+from pandas.io import (
+    formats as formats,
+    json as json,
+    stata as stata,
+)

pandas-stubs/io/formats/__init__.pyi

@@ -0,0 +1 @@
+from pandas.io.formats import style as style

pandas-stubs/io/json/__init__.pyi

@@ -1,2 +1,7 @@
-from pandas.io.json._json import read_json as read_json
+from pandas.io.json._json import (
+    read_json as read_json,
+)
+
+# below are untyped imports so commented out
+# to_json as to_json,; ujson_dumps as ujson_dumps,; ujson_loads as ujson_loads,
 from pandas.io.json._table_schema import build_table_schema as build_table_schema