Code update

PiperOrigin-RevId: 624416121
google-research · Apr 15, 2024 · cc1729f · cc1729f
1 parent 11d4d88
commit cc1729f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,10 +22,12 @@ dependencies = [
     "absl-py",
     "attrs",
     "einops",
+    "fancyflags",
     "jax",
     "numpy",
     "scipy",
     "six",
+    "swirl-dynamics@git+https://github.com/google-research/swirl-dynamics",
     "tensorflow",
 ]