islpy 2025.2.1__cp313-cp313-macosx_11_0_arm64.whl → 2025.2.4__cp313-cp313-macosx_11_0_arm64.whl

islpy/__init__.py

@@ -21,7 +21,7 @@ THE SOFTWARE.
 """
 
 from collections.abc import Collection, Sequence
-from typing import Literal, TypeAlias, TypeVar
+from typing import Literal, TypeAlias, TypeVar, cast
 
 from islpy.version import VERSION, VERSION_TEXT
 
@@ -135,6 +135,7 @@ Alignable: TypeAlias = (
     | Aff | PwAff
 )
 AlignableT = TypeVar("AlignableT", bound=Alignable)
+AlignableT2 = TypeVar("AlignableT2", bound=Alignable)
 
 # }}}
 
@@ -149,44 +150,44 @@ def _get_default_context() -> Context:
     return DEFAULT_CONTEXT
 
 
-def _back_to_basic(new_obj, old_obj):
-    # Work around set_dim_id not being available for Basic{Set,Map}
-    if isinstance(old_obj, BasicSet) and isinstance(new_obj, Set):
-        bsets = new_obj.get_basic_sets()
-
-        if len(bsets) == 0:
-            bset = BasicSet.universe(new_obj.space).complement()
+def _set_dim_id(obj: AlignableT, dt: dim_type, idx: int, id: Id) -> AlignableT:
+    if isinstance(obj, BasicSet):
+        s = obj.to_set().set_dim_id(dt, idx, id)
+        basicsets = s.get_basic_sets()
+        if not basicsets:
+            result = BasicSet.empty(s.space)
         else:
-            bset, = bsets
-
-        return bset
-
-    if isinstance(old_obj, BasicMap) and isinstance(new_obj, Map):
-        bmaps = new_obj.get_basic_maps()
-
-        if len(bmaps) == 0:
-            bmap = BasicMap.universe(new_obj.space).complement()
+            result, = basicsets
+        return cast("AlignableT", result)
+    elif isinstance(obj, BasicMap):
+        m = obj.to_map().set_dim_id(dt, idx, id)
+        basicmaps = m.get_basic_maps()
+        if not basicmaps:
+            result = BasicMap.empty(m.space)
         else:
-            bmap, = bmaps
+            result, = basicmaps
+        return cast("AlignableT", result)
 
-        return bmap
-
-    return new_obj
-
-
-def _set_dim_id(obj, dt, idx, id):
-    return _back_to_basic(obj.set_dim_id(dt, idx, id), obj)
+    return cast("AlignableT", obj.set_dim_id(dt, idx, id))
 
 
 def _align_dim_type(
             template_dt: dim_type,
             obj: AlignableT,
-            template: AlignableT,
+            template: Alignable,
             obj_bigger_ok: bool,
             obj_names: Collection[str],
             template_names: Collection[str],
         ) -> AlignableT:
 
+    # convert to a type that has get_dim_id
+    if isinstance(template, BasicSet):
+        template = template.to_set()
+    elif isinstance(template, BasicMap):
+        template = template.to_map()
+    elif isinstance(template, Aff):
+        template = template.to_pw_aff()
+
     # {{{ deal with Aff, PwAff
 
     # The technique below will not work for PwAff et al, because there is *only*
@@ -314,8 +315,8 @@ def align_spaces(
 
 def align_two(
             obj1: AlignableT,
-            obj2: AlignableT,
-        ) -> tuple[AlignableT, AlignableT]:
+            obj2: AlignableT2,
+        ) -> tuple[AlignableT, AlignableT2]:
     """Align the spaces of two objects, potentially modifying both of them.
 
     See also :func:`align_spaces`.

islpy/_isl.pyi

@@ -2151,6 +2151,8 @@ class FixedBox:
 class Aff:
     def __init__(self, s: str, context: Context | None = None) -> None: ...
 
+    def to_pw_aff(self) -> PwAff: ...
+
     @staticmethod
     def zero_on_domain_space(space: Space) -> Aff: ...
 
@@ -4931,6 +4933,10 @@ class LocalSpace:
 class BasicSet:
     def __init__(self, s: str, context: Context | None = None) -> None: ...
 
+    def __hash__(self) -> int: ...
+
+    def is_params(self) -> bool: ...
+
     def n_dim(self) -> int: ...
 
     def n_param(self) -> int: ...
@@ -5311,8 +5317,6 @@ class BasicSet:
 
     def involves_locals(self) -> bool: ...
 
-    def is_params(self) -> bool: ...
-
     @overload
     def is_strict_subset(self, set2: Set | BasicSet) -> bool: ...
 
@@ -5365,8 +5369,6 @@ class BasicSet:
 
     def plain_is_disjoint(self, set2: Set | BasicSet) -> bool: ...
 
-    def __hash__(self) -> int: ...
-
     def n_basic_set(self) -> int: ...
 
     def foreach_basic_set(self, fn: Callable[[BasicSet], None]) -> None: ...
@@ -5459,13 +5461,13 @@ class BasicSet:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.set_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.set_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.set_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.set_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -5487,15 +5489,15 @@ class BasicSet:
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.bset_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.bset_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.set_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.set_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.set_sub
 
     get_constraints = islpy._monkeypatch.basic_obj_get_constraints
 
@@ -5504,6 +5506,8 @@ class BasicSet:
 class BasicMap:
     def __init__(self, s: str, context: Context | None = None) -> None: ...
 
+    def __hash__(self) -> int: ...
+
     def to_map(self) -> Map: ...
 
     def total_dim(self) -> int: ...
@@ -6108,8 +6112,6 @@ class BasicMap:
 
     def plain_is_equal(self, map2: Map | BasicMap) -> bool: ...
 
-    def __hash__(self) -> int: ...
-
     def n_basic_map(self) -> int: ...
 
     def foreach_basic_map(self, fn: Callable[[BasicMap], None]) -> None: ...
@@ -6220,13 +6222,13 @@ class BasicMap:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.map_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.map_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.map_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.map_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -6246,15 +6248,15 @@ class BasicMap:
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.bmap_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.bmap_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.map_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.map_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.map_sub
 
     get_constraints = islpy._monkeypatch.basic_obj_get_constraints
 
@@ -6724,13 +6726,13 @@ class Set:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.set_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.set_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.set_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.set_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -6752,15 +6754,15 @@ class Set:
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.set_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.set_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.set_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.set_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.set_sub
 
     get_basic_sets = islpy._monkeypatch.set_get_basic_sets
 
@@ -7403,13 +7405,13 @@ class Map:
 
     __ne__ = islpy._monkeypatch.obj_ne
 
-    __lt__ = islpy._monkeypatch.obj_lt
+    __lt__ = islpy._monkeypatch.map_lt
 
-    __le__ = islpy._monkeypatch.obj_le
+    __le__ = islpy._monkeypatch.map_le
 
-    __gt__ = islpy._monkeypatch.obj_gt
+    __gt__ = islpy._monkeypatch.map_gt
 
-    __ge__ = islpy._monkeypatch.obj_ge
+    __ge__ = islpy._monkeypatch.map_ge
 
     project_out_except = islpy._monkeypatch.obj_project_out_except
 
@@ -7429,15 +7431,15 @@ class Map:
 
     __repr__ = islpy._monkeypatch.generic_repr
 
-    __or__ = islpy._monkeypatch.obj_or
+    __and__ = islpy._monkeypatch.map_and
 
-    __ror__ = islpy._monkeypatch.obj_or
+    __rand__ = islpy._monkeypatch.map_and
 
-    __and__ = islpy._monkeypatch.obj_and
+    __or__ = islpy._monkeypatch.map_or
 
-    __rand__ = islpy._monkeypatch.obj_and
+    __ror__ = islpy._monkeypatch.map_or
 
-    __sub__ = islpy._monkeypatch.obj_sub
+    __sub__ = islpy._monkeypatch.map_sub
 
     get_basic_maps = islpy._monkeypatch.map_get_basic_maps
 

islpy/_monkeypatch.py

@@ -5,7 +5,6 @@ from functools import update_wrapper
 from sys import intern
 from typing import (
     TYPE_CHECKING,
-    Any,
     ClassVar,
     Concatenate,
     Literal,
@@ -43,6 +42,10 @@ _CHECK_DIM_TYPES: tuple[_isl.dim_type, ...] = (
 
 # {{{ typing helpers
 
+T = TypeVar("T")
+P = ParamSpec("P")
+ResultT = TypeVar("ResultT")
+
 SelfT = TypeVar("SelfT")
 
 BasicT = TypeVar("BasicT", _isl.BasicSet, _isl.BasicMap)
@@ -54,36 +57,28 @@ ExprLikeT = TypeVar("ExprLikeT", _isl.Aff, _isl.PwAff,
                    _isl.QPolynomial, _isl.PwQPolynomial
                )
 SetLikeT = TypeVar("SetLikeT", bound=_isl.BasicSet | _isl.Set)
+
+SetOrBasic: TypeAlias = _isl.BasicSet | _isl.Set
+SetOrBasicT = TypeVar("SetOrBasicT", bound=SetOrBasic)
+
+MapOrBasic: TypeAlias = _isl.BasicMap | _isl.Map
+MapOrBasicT = TypeVar("MapOrBasicT", bound=MapOrBasic)
+
 SetOrMap: TypeAlias = _isl.BasicSet | _isl.Set | _isl.BasicMap | _isl.Map
-SetOrMapT = TypeVar("SetOrMapT", bound=SetOrMap)
+SetOrMapT = TypeVar("SetOrMapT", _isl.BasicSet, _isl.Set, _isl.BasicMap, _isl.Map)
 
 HasSpace: TypeAlias = (
     _isl.Space
-    | _isl.Aff
-    | _isl.BasicMap
-    | _isl.BasicSet
     | _isl.Constraint
     | _isl.LocalSpace
-    | _isl.Map
+    | _isl.Aff
     | _isl.MultiAff
-    | _isl.MultiId
-    | _isl.MultiPwAff
-    | _isl.MultiUnionPwAff
-    | _isl.MultiVal
-    | _isl.Point
     | _isl.PwAff
     | _isl.PwMultiAff
-    | _isl.PwQPolynomial
-    | _isl.PwQPolynomialFold
-    | _isl.QPolynomial
-    | _isl.QPolynomialFold
+    | _isl.BasicMap
+    | _isl.BasicSet
     | _isl.Set
-    | _isl.UnionMap
-    | _isl.UnionPwAff
-    | _isl.UnionPwMultiAff
-    | _isl.UnionPwQPolynomial
-    | _isl.UnionPwQPolynomialFold
-    | _isl.UnionSet
+    | _isl.Map
     )
 
 
@@ -101,34 +96,30 @@ class IslObject(Protocol):
 
 # {{{ copied verbatim from pytools to avoid numpy/pytools dependency
 
-F = TypeVar("F", bound=Callable[..., Any])
-
-
 class _HasKwargs:
     pass
 
 
-def _memoize_on_first_arg(function: F, cache_dict_name: str | None = None) -> F:
+def _memoize_on_first_arg(
+        function: Callable[Concatenate[T, P], ResultT], *,
+        cache_dict_name: str | None = None) -> Callable[Concatenate[T, P], ResultT]:
     """Like :func:`memoize_method`, but for functions that take the object
     in which do memoization information is stored as first argument.
 
     Supports cache deletion via ``function_name.clear_cache(self)``.
     """
-    from sys import intern
 
     if cache_dict_name is None:
         cache_dict_name = intern(
                 f"_memoize_dic_{function.__module__}{function.__name__}"
                 )
 
-    def wrapper(obj, *args, **kwargs):
-        if kwargs:
-            key = (_HasKwargs, frozenset(kwargs.items()), *args)
-        else:
-            key = args
+    def wrapper(obj: T, *args: P.args, **kwargs: P.kwargs) -> ResultT:
+        key = (_HasKwargs, frozenset(kwargs.items()), *args) if kwargs else args
 
+        assert cache_dict_name is not None
         try:
-            return getattr(obj, cache_dict_name)[key]
+            return cast("ResultT", getattr(obj, cache_dict_name)[key])
         except AttributeError:
             attribute_error = True
         except KeyError:
@@ -138,11 +129,10 @@ def _memoize_on_first_arg(function: F, cache_dict_name: str | None = None) -> F:
         if attribute_error:
             object.__setattr__(obj, cache_dict_name, {key: result})
             return result
-        else:
-            getattr(obj, cache_dict_name)[key] = result
-            return result
+        getattr(obj, cache_dict_name)[key] = result
+        return result
 
-    def clear_cache(obj):
+    def clear_cache(obj: object):
         object.__delattr__(obj, cache_dict_name)
 
     from functools import update_wrapper
@@ -152,7 +142,8 @@ def _memoize_on_first_arg(function: F, cache_dict_name: str | None = None) -> F:
     # into the function's dict is moderately sketchy.
     new_wrapper.clear_cache = clear_cache  # type: ignore[attr-defined]
 
-    return cast("F", new_wrapper)
+    return new_wrapper
+
 
 # }}}
 
@@ -190,7 +181,7 @@ def context_ne(self: object, other: object) -> bool:
     return not self.__eq__(other)
 
 
-def generic_reduce(self: IslObject):
+def generic_reduce(self: HasSpace):
     ctx = self.get_ctx()
     prn = _isl.Printer.to_str(ctx)
     prn = getattr(prn, f"print_{self._base_name}")(self)
@@ -331,21 +322,100 @@ def space_create_from_names(
     return result
 
 
-def obj_or(self: SetOrMapT, other: object) -> SetOrMapT:
+def set_or(
+            self: _isl.Set,
+            other: _isl.Set | _isl.BasicSet,
+        ) -> _isl.Set:
+    try:
+        return self.union(other)
+    except TypeError:
+        return NotImplemented
+
+
+def bset_and(
+            self: _isl.BasicSet,
+            other: SetOrBasicT,
+        ) -> SetOrBasicT:
+    if isinstance(other, _isl.Set):
+        try:
+            return self.to_set().intersect(other)
+        except TypeError:
+            return NotImplemented
+    else:
+        try:
+            return self.intersect(other)
+        except TypeError:
+            return NotImplemented
+
+
+def set_and(
+            self: _isl.Set,
+            other: _isl.Set | _isl.BasicSet,
+        ) -> _isl.Set:
+    if isinstance(self, _isl.BasicSet):
+        self = self.to_set()
+    try:
+        return self.intersect(other)
+    except TypeError:
+        return NotImplemented
+
+
+def set_sub(
+            self: _isl.Set | _isl.BasicSet,
+            other: _isl.Set | _isl.BasicSet,
+        ) -> _isl.Set:
+    if isinstance(self, _isl.BasicSet):
+        self = self.to_set()
+    try:
+        return self.subtract(other)
+    except TypeError:
+        return NotImplemented
+
+
+def map_or(
+            self: _isl.Map | _isl.BasicMap,
+            other: _isl.Map | _isl.BasicMap,
+        ) -> _isl.Map:
+    if isinstance(self, _isl.BasicMap):
+        self = self.to_map()
     try:
         return self.union(other)
     except TypeError:
         return NotImplemented
 
 
-def obj_and(self: SetOrMapT, other: object) -> SetOrMapT:
+def bmap_and(
+            self: _isl.BasicMap,
+            other: MapOrBasicT,
+        ) -> MapOrBasicT:
+    if isinstance(other, _isl.Map):
+        try:
+            return self.to_map().intersect(other)
+        except TypeError:
+            return NotImplemented
+    else:
+        try:
+            return self.intersect(other)
+        except TypeError:
+            return NotImplemented
+
+
+def map_and(
+            self: _isl.Map,
+            other: _isl.Map | _isl.BasicMap,
+        ) -> _isl.Map:
     try:
         return self.intersect(other)
     except TypeError:
         return NotImplemented
 
 
-def obj_sub(self: SetOrMapT, other: object) -> SetOrMapT:
+def map_sub(
+            self: _isl.Map | _isl.BasicMap,
+            other: _isl.Map | _isl.BasicMap,
+        ) -> _isl.Map:
+    if isinstance(self, _isl.BasicMap):
+        self = self.to_map()
     try:
         return self.subtract(other)
     except TypeError:
@@ -538,19 +608,23 @@ def obj_get_var_dict(
 def obj_get_var_ids(
             self: HasSpace,
             dimtype: _isl.dim_type
-        ) -> Sequence[str]:
+        ) -> Sequence[str | None]:
     """Return a list of :class:`Id` instances for :class:`dim_type` *dimtype*."""
-    return [self.get_dim_name(dimtype, i) for i in range(self.dim(dimtype))]
+    return [
+        self.get_dim_name(dimtype, i)
+        for i in range(self.dim(dimtype))]
 
 
 @_memoize_on_first_arg
-def obj_get_var_names(self: HasSpace, dimtype: _isl.dim_type) -> Sequence[str]:
+def obj_get_var_names(self: HasSpace, dimtype: _isl.dim_type) -> Sequence[str | None]:
     """Return a list of dim names (in order) for :class:`dim_type` *dimtype*."""
     return [self.get_dim_name(dimtype, i)
         for i in range(self.dim(dimtype))]
 
 
 def pwaff_get_pieces(self: _isl.PwAff | _isl.Aff) -> list[tuple[_isl.Set, _isl.Aff]]:
+    if isinstance(self, _isl.Aff):
+        self = self.to_pw_aff()
     result: list[tuple[_isl.Set, _isl.Aff]] = []
 
     def append_tuple(s: _isl.Set, v: _isl.Aff):
@@ -583,7 +657,7 @@ def pw_get_aggregate_domain(self: _isl.PwAff | _isl.PwQPolynomial) -> _isl.Set:
 
     result = _isl.Set.empty(self.get_domain_space())
     for dom, _ in self.get_pieces():
-        result = result.union(cast("_isl.Set", dom))
+        result = result.union(dom)
 
     return result
 
@@ -655,7 +729,7 @@ def expr_like_add(self: ExprLikeT, other: ExprLikeT | int | _isl.Val) -> ExprLik
         return NotImplemented
 
 
-def expr_like_sub(self: ExprLikeT, other: ExprLikeT | int | _isl.Val):
+def expr_like_sub(self: ExprLikeT, other: ExprLikeT | int | _isl.Val) -> ExprLikeT:
     if not isinstance(other, ExprLike):
         other = _number_to_expr_like(self, other)
 
@@ -725,29 +799,45 @@ for cls in ALL_CLASSES:
         cls.__ne__ = obj_ne
 
 
-def obj_lt(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_lt(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
     return self.is_strict_subset(other)
 
 
-def obj_le(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_le(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
     return self.is_subset(other)
 
 
-def obj_gt(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_gt(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
     return other.is_strict_subset(self)
 
 
-def obj_ge(self: SetOrMapT, other: SetOrMapT) -> bool:
+def set_ge(self: _isl.BasicSet | _isl.Set, other: _isl.BasicSet | _isl.Set) -> bool:
+    return other.is_subset(self)
+
+
+def map_lt(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
+    return self.is_strict_subset(other)
+
+
+def map_le(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
+    return self.is_subset(other)
+
+
+def map_gt(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
+    return other.is_strict_subset(self)
+
+
+def map_ge(self: _isl.BasicMap | _isl.Map, other: _isl.BasicMap | _isl.Map) -> bool:
     return other.is_subset(self)
 
 
 # {{{ project_out_except
 
 def obj_project_out_except(
-            obj: SetLikeT,
+            obj: SetOrMapT,
             names: Collection[str],
             types: Collection[_isl.dim_type]
-        ) -> SetLikeT:
+        ) -> SetOrMapT:
     """
     :param types: list of :class:`dim_type` determining
         the types of axes to project out
@@ -784,10 +874,10 @@ def obj_project_out_except(
 # {{{ eliminate_except
 
 def obj_eliminate_except(
-            obj: SetLikeT,
+            obj: SetOrMapT,
             names: Collection[str],
             types: Collection[_isl.dim_type]
-        ) -> SetLikeT:
+        ) -> SetOrMapT:
     """
     :param types: list of :class:`dim_type` determining
         the types of axes to eliminate
@@ -869,12 +959,23 @@ def _add_functionality() -> None:
 
     # {{{ Python set-like behavior
 
-    for cls in [_isl.BasicSet, _isl.BasicMap, _isl.Set, _isl.Map]:
-        cls.__or__ = obj_or
-        cls.__ror__ = obj_or
-        cls.__and__ = obj_and
-        cls.__rand__ = obj_and
-        cls.__sub__ = obj_sub
+    _isl.BasicSet.__and__ = bset_and
+    _isl.BasicSet.__rand__ = bset_and
+    _isl.Set.__and__ = set_and
+    _isl.Set.__rand__ = set_and
+    for cls in [_isl.BasicSet, _isl.Set]:
+        cls.__or__ = set_or
+        cls.__ror__ = set_or
+        cls.__sub__ = set_sub
+
+    _isl.BasicMap.__and__ = bmap_and
+    _isl.BasicMap.__rand__ = bmap_and
+    _isl.Map.__and__ = map_and
+    _isl.Map.__rand__ = map_and
+    for cls in [_isl.BasicMap, _isl.Map]:
+        cls.__or__ = map_or
+        cls.__ror__ = map_or
+        cls.__sub__ = map_sub
 
     # }}}
 
@@ -1003,11 +1104,17 @@ for cls in ALL_CLASSES:
 
     # {{{ rich comparisons
 
-    for cls in [_isl.BasicSet, _isl.BasicMap, _isl.Set, _isl.Map]:
-        cls.__lt__ = obj_lt
-        cls.__le__ = obj_le
-        cls.__gt__ = obj_gt
-        cls.__ge__ = obj_ge
+    for cls in [_isl.BasicSet, _isl.Set]:
+        cls.__lt__ = set_lt
+        cls.__le__ = set_le
+        cls.__gt__ = set_gt
+        cls.__ge__ = set_ge
+
+    for cls in [_isl.BasicMap, _isl.Map]:
+        cls.__lt__ = map_lt
+        cls.__le__ = map_le
+        cls.__gt__ = map_gt
+        cls.__ge__ = map_ge
 
     # }}}
 
@@ -1022,10 +1129,6 @@ for cls in ALL_CLASSES:
 _add_functionality()
 
 
-P = ParamSpec("P")
-ResultT = TypeVar("ResultT")
-
-
 _DOWNCAST_RE = re.compile(
           r"Downcast from :class:`([A-Za-z]+)` to :class:`([A-Za-z]+)`.")
 

{islpy-2025.2.1.dist-info → islpy-2025.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: islpy
-Version: 2025.2.1
+Version: 2025.2.4
 Summary: Wrapper around isl, an integer set library
 Author-Email: Andreas Kloeckner <inform@tiker.net>
 License-Expression: MIT

islpy-2025.2.4.dist-info/RECORD

@@ -0,0 +1,9 @@
+islpy/version.py,sha256=9KVEg7OX9uCaA6ElUMeiAh-zDrxsy3u2H6VQm_-LTUE,301
+islpy/_isl.cpython-313-darwin.so,sha256=QhUZVDAMG6d-lfiVJt9fk00yG84tGmf2akrNbgTHu9U,7927080
+islpy/__init__.py,sha256=0AdPhWQMf1DUB5DDisLP39rN0djxueftEWqfX4_R5Hk,13363
+islpy/_isl.pyi,sha256=OI3MI4rqq4XZ3CTXTbmS__lClCMynkL9MxERZ-KgWLk,260125
+islpy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+islpy/_monkeypatch.py,sha256=80xulld5N-jIJbWLGBYSuPnO1CG1N-Z3jVgiNRkkJ9U,34216
+islpy-2025.2.4.dist-info/RECORD,,
+islpy-2025.2.4.dist-info/WHEEL,sha256=LDEcERf6NwqnVU7GVh2ePzziJ1dorOEI8grnlWj9zQ0,141
+islpy-2025.2.4.dist-info/METADATA,sha256=bCRATmwgpiAGBbVa21X7Cqg9fPZNkdFDnuXO4_80yn8,2966

islpy-2025.2.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-islpy-2025.2.1.dist-info/RECORD,,
-islpy-2025.2.1.dist-info/WHEEL,sha256=LDEcERf6NwqnVU7GVh2ePzziJ1dorOEI8grnlWj9zQ0,141
-islpy-2025.2.1.dist-info/METADATA,sha256=rKg9uyDgpaAp2mbvRv2YzFS9seQzhA1jjw75gcTmQvg,2966
-islpy/version.py,sha256=9KVEg7OX9uCaA6ElUMeiAh-zDrxsy3u2H6VQm_-LTUE,301
-islpy/_isl.cpython-313-darwin.so,sha256=3-IElDfITHXIV4NEFhJl2hkRfpvmsopd7CPqCpT4V7E,7927080
-islpy/__init__.py,sha256=6In7YFiBLKNNVIGkpKMiB1H_-H2OQ5ovlqJ_Tz0nAvA,13073
-islpy/_isl.pyi,sha256=dAYJ9hZ0HrpMnUHdruE6cOvKU_Px2kZ7qr6YFDZ-Z-0,260082
-islpy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-islpy/_monkeypatch.py,sha256=NrSQHBRxIVzSuYa2qR_oSnxRoJxVbu-l_Sv1AZcvE0Q,31242