Merge pull request #724 from mlcommons/wandb_version_fix

Dependency fix
mlcommons · Mar 26, 2024 · 060b20d · 060b20d
2 parents ec31805 + 6bc5e9c
commit 060b20d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -120,6 +120,7 @@ jax_core_deps =
   # upgrade jax.
   chex==0.1.7
   ml_dtypes==0.2.0
+  protobuf==4.25.3
 
 # JAX CPU
 jax_cpu =