Merge pull request #29 from invrs-io/compile

Check bounds during compile

.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "v0.6.1"
+current_version = "v0.6.2"
 commit = true
 commit_args = "--no-verify"
 tag = true

README.md

@@ -1,5 +1,5 @@
 # totypes - Custom types for topology optimization
-`v0.6.1`
+`v0.6.2`
 
 ## Overview
 

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 
 name = "totypes"
-version = "v0.6.1"
+version = "v0.6.2"
 description = "Custom datatypes useful in a topology optimization context"
 keywords = ["topology", "optimization", "jax", "inverse design"]
 readme = "README.md"

src/totypes/__init__.py

@@ -3,7 +3,7 @@
 Copyright (c) 2023 The INVRS-IO authors.
 """
 
-__version__ = "v0.6.1"
+__version__ = "v0.6.2"
 __author__ = "Martin F. Schubert <[email protected]>"
 
 __all__ = ["json_utils", "symmetry", "types"]

src/totypes/types.py

@@ -68,14 +68,15 @@ def __post_init__(self) -> None:
         if self.upper_bound is not None and self.lower_bound is not None:
             invalid_bounds = self.lower_bound >= self.upper_bound
             is_array = isinstance(invalid_bounds, (jnp.ndarray, onp.ndarray))
-            if (
-                is_array
-                and invalid_bounds.any()  # type: ignore[union-attr]
-                or (not is_array and invalid_bounds)
-            ):
-                raise ValueError(
-                    "`upper_bound` must be strictly greater than `lower_bound`."
-                )
+            with jax.ensure_compile_time_eval():
+                if (
+                    is_array
+                    and invalid_bounds.any()  # type: ignore[union-attr]
+                    or (not is_array and invalid_bounds)
+                ):
+                    raise ValueError(
+                        "`upper_bound` must be strictly greater than `lower_bound`."
+                    )
 
     @property
     def shape(self) -> Tuple[int, ...]: