From 03c71634e677c63300b32de8a35e0db061cd0807 Mon Sep 17 00:00:00 2001 From: siriuz42 Date: Thu, 11 Jul 2024 18:53:29 +0000 Subject: [PATCH] fixed dependencies for PyPI --- pyproject.toml | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 45865df..91e1605 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,17 +1,20 @@ # This project can be installed with `python3 -m pip install -e .` from the main directory. [project] -name = "timesfm" +name = "timesfm-jax" description = "Open weights time-series foundation model from Google Research." -version = "1.0.0" +version = "1.0.1" dependencies = [ - "einshape>=1.0.0", - "paxml>=1.4.0", - "praxis>=1.4.0", - "jax>=0.4.26", - "numpy>=1.26.4", - "pandas>=2.1.4", - "scikit-learn>=1.5.1", + "jax==0.4.26", + "paxml==1.4.0", + "praxis==1.4.0", + "jaxlib==0.4.26", + "numpy==1.26.4", + "pandas==2.1.4", + "einshape==1.0.0", + "utilsforecast==0.1.10", + "huggingface_hub[cli]==0.23.0", + "scikit-learn==1.5.1", ] authors = [ {name = "Rajat Sen", email = "senrajat@google.com"},