Merge pull request #724 from mlcommons/wandb_version_fix

priyakasimbeg · web-flow · commit 060b20d1aef9 · 2024-03-25T19:11:41.000-07:00
Dependency fix
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 =
