Pin JAX version to less than v0.6.0 due to breaking changes (#300)

matt-graham · web-flow · commit 351fc94835ed · 2025-04-23T14:36:41.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",
     "torch",
 ]

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`	`"torch",`
`34`	`34`	`]`