get_constraints: fix space handling

Author: inducer

namedisl/set_like.py

@@ -344,19 +344,16 @@ def remove_redundancies(self):
 @dataclass(frozen=True, eq=False)
 class _NamedIslBasic(_NamedIslSetOrMapLike[IslBasicT_co]):
     __doc__ = """
-    .. automethod:: get_constraints
     """
 
-    def get_constraints(self):
-        return [Constraint(cns, self.space) for cns in self._obj.get_constraints()]
-
 
 @dataclass(frozen=True, eq=False)
 class BasicSet(_NamedIslSetLike[isl.BasicSet], _NamedIslBasic[isl.BasicSet]):
     __doc__ = f"""
     .. automethod:: add_constraint
     .. autoattribute:: affs
     .. automethod:: as_set
+    .. automethod:: get_constraints
     {_NamedIslSetLike.__doc__}
     {_NamedIslBasic.__doc__}
     {_NamedIslSetOrMapLike.__doc__}
@@ -370,6 +367,11 @@ def add_constraint(self, cns: Constraint, /) -> BasicSet:
                 raise ValueError("spaces don't match")
         return BasicSet(self._obj.add_constraint(cns._obj), self.space)
 
+    def get_constraints(self):
+        return [
+            Constraint(cns, self.space.with_empty_dim_type(DimType.in_))
+            for cns in self._obj.get_constraints()]
+
     @cached_property
     def affs(self) -> Mapping[str | Literal[0], Aff]:
         r"""
@@ -508,6 +510,7 @@ class BasicMap(_NamedIslMapLike[isl.BasicMap], _NamedIslBasic[isl.BasicMap]):
     .. automethod:: range
     .. automethod:: intersect_domain
     .. automethod:: intersect_range
+    .. automethod:: get_constraints
     {_NamedIslMapLike.__doc__}
     {_NamedIslBasic.__doc__}
     {_NamedIslSetOrMapLike.__doc__}
@@ -519,6 +522,10 @@ def add_constraint(self, cns: Constraint, /) -> BasicMap:
                 raise ValueError("spaces don't match")
         return BasicMap(self._obj.add_constraint(cns._obj), self.space)
 
+    def get_constraints(self):
+        return [
+            Constraint(cns, self.space) for cns in self._obj.get_constraints()]
+
     def domain(self) -> BasicSet:
         return BasicSet(
             self._obj.domain(),