fix: Conditionally import typing-extensions (#91)

Co-authored-by: Oliver Borchert <[email protected]>

Author: jhall-bp

dataframely/_base_collection.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 
+import sys
 import textwrap
 import typing
 from abc import ABCMeta
@@ -11,13 +12,17 @@
 from typing import Annotated, Any, cast, get_args, get_origin
 
 import polars as pl
-from typing_extensions import Self
 
 from ._filter import Filter
 from ._typing import LazyFrame as TypedLazyFrame
 from .exc import AnnotationImplementationError, ImplementationError
 from .schema import Schema
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 _MEMBER_ATTR = "__dataframely_members__"
 _FILTER_ATTR = "__dataframely_filters__"
 

dataframely/_base_schema.py

@@ -3,19 +3,24 @@
 
 from __future__ import annotations
 
+import sys
 import textwrap
 from abc import ABCMeta
 from copy import copy
 from dataclasses import dataclass, field
 from typing import Any
 
 import polars as pl
-from typing_extensions import Self
 
 from ._rule import GroupRule, Rule
 from .columns import Column
 from .exc import ImplementationError
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 _COLUMN_ATTR = "__dataframely_columns__"
 _RULE_ATTR = "__dataframely_rules__"
 

dataframely/_rule.py

@@ -3,12 +3,17 @@
 
 from __future__ import annotations
 
+import sys
 from collections import defaultdict
 from collections.abc import Callable
 from typing import Any
 
 import polars as pl
-from typing_extensions import Self
+
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
 
 ValidationFunction = Callable[[], pl.Expr]
 

dataframely/collection.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: BSD-3-Clause
 
 import json
+import sys
 import warnings
 from abc import ABC
 from collections.abc import Mapping, Sequence
@@ -11,7 +12,6 @@
 
 import polars as pl
 import polars.exceptions as plexc
-from typing_extensions import Self
 
 from ._base_collection import BaseCollection, CollectionMember
 from ._filter import Filter
@@ -33,6 +33,11 @@
 from .random import Generator
 from .schema import _schema_from_dict
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 
 class Collection(BaseCollection, ABC):
     """Base class for all collections of data frames with a predefined schema.

dataframely/columns/_base.py

@@ -4,13 +4,13 @@
 from __future__ import annotations
 
 import inspect
+import sys
 from abc import ABC, abstractmethod
 from collections import Counter
 from collections.abc import Callable
 from typing import Any, TypeAlias, cast
 
 import polars as pl
-from typing_extensions import Self
 
 from dataframely._compat import pa, sa, sa_TypeEngine
 from dataframely._deprecation import (
@@ -20,6 +20,11 @@
 from dataframely._polars import PolarsDataType
 from dataframely.random import Generator
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 Check: TypeAlias = (
     Callable[[pl.Expr], pl.Expr]
     | list[Callable[[pl.Expr], pl.Expr]]

dataframely/columns/_mixins.py

@@ -1,11 +1,11 @@
 # Copyright (c) QuantCo 2025-2025
 # SPDX-License-Identifier: BSD-3-Clause
 
+import sys
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any, Generic, Protocol, TypeVar
 
 import polars as pl
-from typing_extensions import Self
 
 if TYPE_CHECKING:  # pragma: no cover
     from ._base import Column
@@ -14,6 +14,11 @@
 else:
     Base = object
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 # ----------------------------------- ORDINAL MIXIN ---------------------------------- #
 
 

dataframely/columns/array.py

@@ -4,11 +4,11 @@
 from __future__ import annotations
 
 import math
+import sys
 from collections.abc import Sequence
 from typing import Any, Literal, cast
 
 import polars as pl
-from typing_extensions import Self
 
 from dataframely._compat import pa, sa, sa_TypeEngine
 from dataframely.random import Generator
@@ -17,6 +17,11 @@
 from ._registry import column_from_dict, register
 from .struct import Struct
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 
 @register
 class Array(Column):

dataframely/columns/list.py

@@ -3,11 +3,11 @@
 
 from __future__ import annotations
 
+import sys
 from itertools import chain
 from typing import Any, cast
 
 import polars as pl
-from typing_extensions import Self
 
 from dataframely._compat import pa, sa, sa_TypeEngine
 from dataframely._polars import PolarsDataType
@@ -17,6 +17,11 @@
 from ._registry import column_from_dict, register
 from .struct import Struct
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 
 @register
 class List(Column):

dataframely/columns/struct.py

@@ -3,10 +3,10 @@
 
 from __future__ import annotations
 
+import sys
 from typing import Any, cast
 
 import polars as pl
-from typing_extensions import Self
 
 from dataframely._compat import pa, sa, sa_TypeEngine
 from dataframely._polars import PolarsDataType
@@ -15,6 +15,11 @@
 from ._base import Check, Column
 from ._registry import column_from_dict, register
 
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
+
 
 @register
 class Struct(Column):

dataframely/config.py

@@ -2,10 +2,14 @@
 # SPDX-License-Identifier: BSD-3-Clause
 
 import contextlib
+import sys
 from types import TracebackType
 from typing import TypedDict
 
-from typing_extensions import Unpack
+if sys.version_info >= (3, 11):
+    from typing import Unpack
+else:
+    from typing_extensions import Unpack
 
 
 class Options(TypedDict):