Update environments and dependencies.

PiperOrigin-RevId: 632263761
2024-05-09 14:05:01 -07:00
parent a4fd4ca7c3
commit b389495851
6 changed files with 62 additions and 20 deletions
@@ -5,6 +5,7 @@ name = "timesfm"
 description = "Open weights time-series foundation model from Google Research."
 version = "0.0.1"
 dependencies = [
+    "einshape>=1.0.0",
    "paxml>=1.4.0",
    "praxis>=1.4.0",
    "jax>=0.4.26",