Set the test version of flax used in the equivalence test.

PiperOrigin-RevId: 469472269
google-deepmind · Aug 23, 2022 · b5f1b6e · b5f1b6e
1 parent 2d7d6fe
commit b5f1b6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
@@ -1,3 +1,3 @@
 dm-haiku>=0.0.3
 dm-tree>=0.1.7
-flax>=0.3.1
+flax==0.5.3