Pin JAX version to less than v0.6.0 due to breaking changes

matt-graham · matt-graham · commit a91729f36696 · 2025-04-23T11:13:31.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,7 +28,7 @@ classifiers = [
 description = "Differentiable and accelerated spherical transforms with JAX"
 dependencies = [
     "numpy>=1.20",
-    "jax>=0.3.13",
+    "jax>=0.3.13,<0.6.0",
     "jaxlib",
 ]
 dynamic = [

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ classifiers = [`
`28`	`28`	`description = "Differentiable and accelerated spherical transforms with JAX"`
`29`	`29`	`dependencies = [`
`30`	`30`	`"numpy>=1.20",`
`31`		`- "jax>=0.3.13",`
	`31`	`+ "jax>=0.3.13,<0.6.0",`
`32`	`32`	`"jaxlib",`
`33`	`33`	`]`
`34`	`34`	`dynamic = [`