Updating dm-haiku test dependency. We can use numbered version again now that dm-haiku has been updated on PyPI.

james-martens · KfacJaxDev · commit 85535ff52493 · 2025-05-20T10:17:04.000-07:00
PiperOrigin-RevId: 761132560
diff --git a/pyproject.toml b/pyproject.toml
@@ -70,7 +70,7 @@ tests = [  # these should be version pinned?
   "distrax>=0.1.3",
   "jax>=0.6.0",
   "jaxlib>=0.4.27",
-  "dm-haiku @ git+https://github.com/deepmind/dm-haiku@ffd905e113b47d24ce6060fa1fb9f14d3808baf7",
+  "dm-haiku>=0.0.9",
   "dm-tree>=0.1.7",
   "optax>=0.1.4",
 ]

Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ tests = [ # these should be version pinned?`
`70`	`70`	`"distrax>=0.1.3",`
`71`	`71`	`"jax>=0.6.0",`
`72`	`72`	`"jaxlib>=0.4.27",`
`73`		`- "dm-haiku @ git+https://github.com/deepmind/dm-haiku@ffd905e113b47d24ce6060fa1fb9f14d3808baf7",`
	`73`	`+ "dm-haiku>=0.0.9",`
`74`	`74`	`"dm-tree>=0.1.7",`
`75`	`75`	`"optax>=0.1.4",`
`76`	`76`	`]`