Pin to flax>=0.8.0

pangeo-data · Feb 16, 2024 · 0417e79 · 0417e79
1 parent b6f62f6
commit 0417e79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml-notebook/environment.yml b/ml-notebook/environment.yml
@@ -5,7 +5,7 @@ channels:
  - conda-forge
  - nodefaults
 dependencies:
- - flax
+ - flax>=0.8.0
  - jax
  - jupyterlab-nvdashboard
  - keras-cv