Merge branch 'main' into jax_version

Author: EiffL

jax_galsim/fitswcs.py

@@ -835,27 +835,15 @@ def __eq__(self, other):
             and self.center == other.center
             and (
                 (self.pv is None and other.pv is None)
-                or (
-                    self.pv is not None
-                    and other.pv is not None
-                    and jnp.array_equal(self.pv, other.pv)
-                )
+                or jnp.array_equal(self.pv, other.pv)
             )
             and (
                 (self.ab is None and other.ab is None)
-                or (
-                    self.ab is not None
-                    and other.ab is not None
-                    and jnp.array_equal(self.ab, other.ab)
-                )
+                or jnp.array_equal(self.ab, other.ab)
             )
             and (
                 (self.abp is None and other.abp is None)
-                or (
-                    self.abp is not None
-                    and other.abp is not None
-                    and jnp.array_equal(self.abp, other.abp)
-                )
+                or jnp.array_equal(self.abp, other.abp)
             )
         )
 

pyproject.toml

@@ -16,7 +16,7 @@ readme = "README.md"
 dependencies = [
     "numpy >=1.18.0",
     "galsim >=2.3.0",
-    "jax >=0.5.0",
+    "jax >=0.8.0",
     "astropy >=2.0",
     "quadax",
 ]