from typing to typing-extensions

mesa_frames/abstract/agents.py

@@ -3,9 +3,11 @@
 from abc import abstractmethod
 from collections.abc import Callable, Collection, Iterable, Iterator, Sequence
 from contextlib import suppress
-from typing import TYPE_CHECKING, Any, Literal, Self, overload
 
 from numpy.random import Generator
+from typing_extensions import Any, Self, overload
+
+from typing import TYPE_CHECKING, Literal
 
 from mesa_frames.abstract.mixin import CopyMixin
 from mesa_frames.types import BoolSeries, DataFrame, IdsLike, Index, MaskLike, Series

mesa_frames/abstract/mixin.py

@@ -1,6 +1,7 @@
 from abc import ABC, abstractmethod
 from copy import copy, deepcopy
-from typing import Self
+
+from typing_extensions import Self
 
 
 class CopyMixin(ABC):

mesa_frames/concrete/agents.py

@@ -1,10 +1,10 @@
 from collections import defaultdict
-from collections.abc import Callable, Iterable, Iterator, Sequence
-from typing import Any, Literal, Self, cast, overload
-
-from collections.abc import Collection
+from collections.abc import Callable, Collection, Iterable, Iterator, Sequence
 
 import polars as pl
+from typing_extensions import Any, Self, overload
+
+from typing import Literal, cast
 
 from mesa_frames.abstract.agents import AgentContainer, AgentSetDF
 from mesa_frames.types import DataFrame, IdsLike, MaskLike, Series

mesa_frames/concrete/agentset_pandas.py

@@ -1,8 +1,10 @@
 from collections.abc import Callable, Collection, Iterable, Iterator, Sequence
-from typing import TYPE_CHECKING, Any, Self, overload
 
 import pandas as pd
 import polars as pl
+from typing_extensions import Any, Self, overload
+
+from typing import TYPE_CHECKING
 
 from mesa_frames.abstract.agents import AgentSetDF
 from mesa_frames.concrete.agentset_polars import AgentSetPolars

mesa_frames/concrete/agentset_polars.py

@@ -1,8 +1,10 @@
 from collections.abc import Callable, Collection, Iterable, Iterator, Sequence
-from typing import TYPE_CHECKING, Any, Self, overload
 
 import polars as pl
 from polars.type_aliases import IntoExpr
+from typing_extensions import Any, Self, overload
+
+from typing import TYPE_CHECKING
 
 from mesa_frames.concrete.agents import AgentSetDF
 from mesa_frames.types import PolarsIdsLike, PolarsMaskLike

mesa_frames/concrete/model.py

@@ -1,7 +1,9 @@
 from collections.abc import Sequence
-from typing import TYPE_CHECKING, Any
 
 import numpy as np
+from typing_extensions import Any
+
+from typing import TYPE_CHECKING
 
 from mesa_frames.concrete.agents import AgentsDF
 

mesa_frames/types.py

@@ -1,9 +1,9 @@
 from collections.abc import Collection, Hashable
-from typing import Literal
 
 import pandas as pd
 import polars as pl
 from numpy import ndarray
+from typing import Literal
 
 ####----- Agnostic Types -----####
 AgnosticMask = Literal["all", "active"] | Hashable | None

pyproject.toml

@@ -12,6 +12,7 @@ authors = [
 license = { text = "MIT" }
 dependencies = [
   "numpy~=1.26",
+  "typing-extensions>=4.9"
 ]
 
 [project.optional-dependencies]