Rework _T_OBJECT

Author: hamdanal

pandas-stubs/core/indexes/base.pyi

@@ -275,7 +275,7 @@ class Index(IndexOpsMixin[S1]):
         Index[int],
         Index[bytes],
         Index[_str],
-        Index[type[object]],
+        Index,
     ]: ...
     @final
     def is_(self, other) -> bool: ...

pandas-stubs/core/series.pyi

@@ -1181,7 +1181,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         Series[int],
         Series[bytes],
         Series[_str],
-        Series[type[object]],
+        Series,
     ]: ...
     @property
     def dt(self) -> CombinedDatetimelikeProperties: ...

pandas-stubs/core/strings/accessor.pyi

@@ -47,7 +47,7 @@ _T_BYTES = TypeVar("_T_BYTES", bound=Series[bytes] | Index[bytes])
 # Used for the result of str.decode
 _T_STR = TypeVar("_T_STR", bound=Series[str] | Index[str])
 # Used for the result of str.partition
-_T_OBJECT = TypeVar("_T_OBJECT", bound=Series[type[object]] | Index[type[object]])
+_T_OBJECT = TypeVar("_T_OBJECT", bound=Series | Index)
 
 class StringMethods(
     NoNewAttributesMixin,

tests/test_string_accessors.py

@@ -356,12 +356,12 @@ def test_series_overloads_partition():
         assert_type(s.str.partition(sep=";", expand=True), pd.DataFrame), pd.DataFrame
     )
     check(
-        assert_type(s.str.partition(sep=";", expand=False), "pd.Series[type[object]]"),
+        assert_type(s.str.partition(sep=";", expand=False), pd.Series),
         pd.Series,
         object,
     )
     check(
-        assert_type(s.str.partition(expand=False), "pd.Series[type[object]]"),
+        assert_type(s.str.partition(expand=False), pd.Series),
         pd.Series,
         object,
     )
@@ -371,15 +371,11 @@ def test_series_overloads_partition():
         assert_type(s.str.rpartition(sep=";", expand=True), pd.DataFrame), pd.DataFrame
     )
     check(
-        assert_type(s.str.rpartition(sep=";", expand=False), "pd.Series[type[object]]"),
-        pd.Series,
-        object,
-    )
-    check(
-        assert_type(s.str.rpartition(expand=False), "pd.Series[type[object]]"),
+        assert_type(s.str.rpartition(sep=";", expand=False), pd.Series),
         pd.Series,
         object,
     )
+    check(assert_type(s.str.rpartition(expand=False), pd.Series), pd.Series, object)
 
 
 def test_index_overloads_partition():
@@ -400,7 +396,7 @@ def test_index_overloads_partition():
         pd.MultiIndex,
     )
     check(
-        assert_type(idx.str.partition(sep=";", expand=False), "pd.Index[type[object]]"),
+        assert_type(idx.str.partition(sep=";", expand=False), pd.Index),
         pd.Index,
         object,
     )
@@ -411,9 +407,7 @@ def test_index_overloads_partition():
         pd.MultiIndex,
     )
     check(
-        assert_type(
-            idx.str.rpartition(sep=";", expand=False), "pd.Index[type[object]]"
-        ),
+        assert_type(idx.str.rpartition(sep=";", expand=False), pd.Index),
         pd.Index,
         object,
     )
@@ -490,16 +484,12 @@ def test_series_overloads_extract():
         assert_type(s.str.extract(r"[ab](\d)", expand=True), pd.DataFrame), pd.DataFrame
     )
     check(
-        assert_type(
-            s.str.extract(r"[ab](\d)", expand=False), "pd.Series[type[object]]"
-        ),
+        assert_type(s.str.extract(r"[ab](\d)", expand=False), pd.Series),
         pd.Series,
         object,
     )
     check(
-        assert_type(
-            s.str.extract(r"[ab](\d)", re.IGNORECASE, False), "pd.Series[type[object]]"
-        ),
+        assert_type(s.str.extract(r"[ab](\d)", re.IGNORECASE, False), pd.Series),
         pd.Series,
         object,
     )
@@ -513,16 +503,12 @@ def test_index_overloads_extract():
         pd.DataFrame,
     )
     check(
-        assert_type(
-            idx.str.extract(r"[ab](\d)", expand=False), "pd.Index[type[object]]"
-        ),
+        assert_type(idx.str.extract(r"[ab](\d)", expand=False), pd.Index),
         pd.Index,
         object,
     )
     check(
-        assert_type(
-            idx.str.extract(r"[ab](\d)", re.IGNORECASE, False), "pd.Index[type[object]]"
-        ),
+        assert_type(idx.str.extract(r"[ab](\d)", re.IGNORECASE, False), pd.Index),
         pd.Index,
         object,
     )