restrict orbax version

cgarciae · Mar 16, 2023 · 1ad983d · 1ad983d
1 parent ff7bb21
commit 1ad983d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,6 +16,7 @@ einop = "*"
 numpy = "*" 
 # jax-metrics = {path = "../jax_metrics"}
 jax-metrics = "^0.2.1"
+orbax = "<0.1.4"
 
 [tool.poetry.group.test.dependencies]
 pytest = ">=7.1.3"