Fix ipywidgets dependency.

PiperOrigin-RevId: 614832168
google-research · Mar 11, 2024 · 303570b · 303570b
1 parent 9f28ba5
commit 303570b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,6 +31,7 @@ numba = "^0.57"
 librosa = "^0.10.1"
 numpy = "^1.24.0"
 tqdm = "^4.6"
+ipywidgets = "^8.1"
 
 [tool.poetry.group.jaxtrain]
 optional = true