islpy 2025.2.4__tar.gz → 2025.2.5__tar.gz

{islpy-2025.2.4 → islpy-2025.2.5}/PKG-INFO

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

{islpy-2025.2.4 → islpy-2025.2.5}/islpy/_monkeypatch.py

@@ -616,10 +616,34 @@ def obj_get_var_ids(
 
 
 @_memoize_on_first_arg
-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*."""
+def obj_get_var_names_not_none(
+            self: HasSpace,
+            dimtype: _isl.dim_type,
+        ) -> Sequence[str]:
+    """Return a list of dim names (in order) for :class:`dim_type` *dimtype*.
+
+    Raise :exc:`ValueError` if any of the names is *None*.
+
+    .. versionadded:: 2025.2.5
+    """
+    ndim = self.dim(dimtype)
+    res = [n
+        for i in range(ndim)
+        if (n := self.get_dim_name(dimtype, i)) is not None]
+    if len(res) != ndim:
+        raise ValueError("None encountered in dim names")
+    return res
+
+
+@_memoize_on_first_arg
+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))]
+            for i in range(self.dim(dimtype))]
 
 
 def pwaff_get_pieces(self: _isl.PwAff | _isl.Aff) -> list[tuple[_isl.Set, _isl.Aff]]:
@@ -1037,6 +1061,7 @@ for cls in ALL_CLASSES:
         cls.get_var_dict = obj_get_var_dict
         cls.get_var_ids = obj_get_var_ids
         cls.get_var_names = obj_get_var_names
+        cls.get_var_names_not_none = obj_get_var_names_not_none
 
     # }}}
 

{islpy-2025.2.4 → islpy-2025.2.5}/pyproject.toml

@@ -11,7 +11,7 @@ requires = [
 
 [project]
 name = "islpy"
-version = "2025.2.4"
+version = "2025.2.5"
 description = "Wrapper around isl, an integer set library"
 readme = "README.rst"
 license = "MIT"