fix: remove usage of typing_extensions where possible

Author: parthea

packages/bigframes/bigframes/core/utils.py

@@ -16,20 +16,19 @@
 import re
 import typing
 import warnings
-from typing import Hashable, Iterable, List
+from typing import Hashable, Iterable, List, TypeGuard
 
 import bigframes_vendored.pandas.io.common as vendored_pandas_io_common
 import numpy as np
 import pandas as pd
-import typing_extensions
 
 import bigframes.exceptions as bfe
 
 UNNAMED_COLUMN_ID = "bigframes_unnamed_column"
 UNNAMED_INDEX_ID = "bigframes_unnamed_index"
 
 
-def is_gcs_path(value) -> typing_extensions.TypeGuard[str]:
+def is_gcs_path(value) -> TypeGuard[str]:
     return isinstance(value, str) and value.startswith("gs://")
 
 
@@ -43,11 +42,11 @@ def get_axis_number(axis: typing.Union[str, int]) -> typing.Literal[0, 1]:
 
 def is_list_like(
     obj: typing.Any, allow_sets: bool = True
-) -> typing_extensions.TypeGuard[typing.Sequence]:
+) -> TypeGuard[typing.Sequence]:
     return pd.api.types.is_list_like(obj, allow_sets=allow_sets)
 
 
-def is_dict_like(obj: typing.Any) -> typing_extensions.TypeGuard[typing.Mapping]:
+def is_dict_like(obj: typing.Any) -> TypeGuard[typing.Mapping]:
     return pd.api.types.is_dict_like(obj)
 
 

packages/bigframes/bigframes/series.py

@@ -33,6 +33,7 @@
     Optional,
     Sequence,
     Tuple,
+    TypeGuard,
     TypeVar,
     Union,
     cast,
@@ -45,7 +46,6 @@
 import numpy
 import pandas
 import pyarrow as pa
-import typing_extensions
 from pandas.api import extensions as pd_ext
 
 import bigframes.core
@@ -2825,5 +2825,5 @@ def _throw_if_null_index(self, opname: __builtins__.str):
             )
 
 
-def _is_list_like(obj: typing.Any) -> typing_extensions.TypeGuard[typing.Sequence]:
+def _is_list_like(obj: typing.Any) -> TypeGuard[typing.Sequence]:
     return pandas.api.types.is_list_like(obj)

packages/bigframes/third_party/bigframes_vendored/ibis/common/bases.py

@@ -3,14 +3,18 @@
 from __future__ import annotations
 
 import collections.abc
+import sys
 from abc import abstractmethod
 from typing import TYPE_CHECKING, Any
 from weakref import WeakValueDictionary
 
 if TYPE_CHECKING:
     from collections.abc import Mapping
 
-    from typing_extensions import Self
+    if sys.version_info >= (3, 11):
+        from typing import Self
+    else:
+        from typing_extensions import Self
 
 
 class AbstractMeta(type):

packages/bigframes/third_party/bigframes_vendored/ibis/common/collections.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import collections.abc
+import sys
 from abc import abstractmethod
 from itertools import tee
 from typing import TYPE_CHECKING, Any, Generic, TypeVar
@@ -12,7 +13,10 @@
 from public import public
 
 if TYPE_CHECKING:
-    from typing_extensions import Self
+    if sys.version_info >= (3, 11):
+        from typing import Self
+    else:
+        from typing_extensions import Self
 
 K = TypeVar("K", bound=collections.abc.Hashable)
 V = TypeVar("V")

packages/bigframes/third_party/bigframes_vendored/ibis/common/graph.py

@@ -5,6 +5,7 @@
 from __future__ import annotations
 
 import itertools
+import sys
 from abc import abstractmethod
 from collections import deque
 from collections.abc import Callable, Iterable, Iterator, KeysView, Mapping, Sequence
@@ -16,7 +17,10 @@
 from bigframes_vendored.ibis.util import experimental, promote_list
 
 if TYPE_CHECKING:
-    from typing_extensions import Self
+    if sys.version_info >= (3, 11):
+        from typing import Self
+    else:
+        from typing_extensions import Self
 
     N = TypeVar("N")
 

packages/bigframes/third_party/bigframes_vendored/ibis/common/grounds.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import contextlib
+import sys
 from copy import copy
 from typing import Any, ClassVar, Union, get_origin
 
@@ -23,7 +24,11 @@
 from bigframes_vendored.ibis.common.collections import FrozenDict  # noqa: TCH001
 from bigframes_vendored.ibis.common.patterns import Pattern
 from bigframes_vendored.ibis.common.typing import evaluate_annotations
-from typing_extensions import Self, dataclass_transform
+
+if sys.version_info >= (3, 11):
+    from typing import Self, dataclass_transform
+else:
+    from typing_extensions import Self, dataclass_transform
 
 
 class AnnotableMeta(AbstractMeta):

packages/bigframes/third_party/bigframes_vendored/ibis/common/typing.py

@@ -7,26 +7,23 @@
 import sys
 from abc import abstractmethod
 from itertools import zip_longest
-from typing import TYPE_CHECKING, Any, Optional, TypeVar, Union, get_args, get_origin
+from typing import TYPE_CHECKING, Any, Optional, TypeVar, get_args, get_origin
 from typing import get_type_hints as _get_type_hints
 
 from bigframes_vendored.ibis.common.bases import Abstract
 from bigframes_vendored.ibis.common.caching import memoize
 
 if TYPE_CHECKING:
-    from typing_extensions import Self
-
-if sys.version_info >= (3, 10):
-    from types import UnionType
-    from typing import TypeAlias
+    if sys.version_info >= (3, 11):
+        from typing import Self
+    else:
+        from typing_extensions import Self
 
-    # Keep this alias in sync with unittest.case._ClassInfo
-    _ClassInfo: TypeAlias = type | UnionType | tuple["_ClassInfo", ...]
-else:
-    from typing_extensions import TypeAlias
+from types import UnionType
+from typing import TypeAlias
 
-    UnionType = object()
-    _ClassInfo: TypeAlias = Union[type, tuple["_ClassInfo", ...]]
+# Keep this alias in sync with unittest.case._ClassInfo
+_ClassInfo: TypeAlias = type | UnionType | tuple["_ClassInfo", ...]
 
 
 T = TypeVar("T")

packages/bigframes/third_party/bigframes_vendored/ibis/expr/builders.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import math
+import sys
 from typing import TYPE_CHECKING, Any, Literal, Optional, Union
 
 import bigframes_vendored.ibis
@@ -19,7 +20,10 @@
 from bigframes_vendored.ibis.common.typing import VarTuple  # noqa: TCH001
 
 if TYPE_CHECKING:
-    from typing_extensions import Self
+    if sys.version_info >= (3, 11):
+        from typing import Self
+    else:
+        from typing_extensions import Self
 
 
 class Builder(Concrete):

packages/bigframes/third_party/bigframes_vendored/ibis/expr/datatypes/core.py

@@ -5,6 +5,7 @@
 import datetime as pydatetime
 import decimal as pydecimal
 import numbers
+import sys
 import uuid as pyuuid
 from abc import abstractmethod
 from collections.abc import Iterable, Iterator, Mapping, Sequence
@@ -29,7 +30,11 @@
 from bigframes_vendored.ibis.common.patterns import Coercible, CoercionError
 from bigframes_vendored.ibis.common.temporal import IntervalUnit, TimestampUnit
 from public import public
-from typing_extensions import Self
+
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
 
 
 @lazy_singledispatch

packages/bigframes/third_party/bigframes_vendored/ibis/expr/operations/core.py

@@ -3,8 +3,9 @@
 from __future__ import annotations
 
 from abc import abstractmethod
-from typing import Generic, Optional
+from typing import Any, Generic, Optional, TypeVar
 
+import sys
 import bigframes_vendored.ibis.expr.datashape as ds
 import bigframes_vendored.ibis.expr.datatypes as dt
 import bigframes_vendored.ibis.expr.rules as rlz
@@ -15,7 +16,11 @@
 from bigframes_vendored.ibis.common.typing import DefaultTypeVars
 from bigframes_vendored.ibis.util import is_iterable
 from public import public
-from typing_extensions import Any, Self, TypeVar
+
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
 
 
 @public