Github action poetry piblish, poetry

.github/workflows/main.yml

@@ -0,0 +1,29 @@
+
+name: Poetry publish
+
+on:
+  push:
+    branches: [ "master" ]
+  pull_request:
+    branches: [ "master" ]
+
+jobs:
+  buildPush:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: '3.10'
+      - name: Install Poetry
+        run: |
+          python -m pip install --upgrade pip
+          pip install poetry
+      - name: Set Version number
+        run: |
+          poetry version ${{ github.event_name }}
+      - name: Build and Publish to PyPI
+        run: |
+          poetry config pypi-token.pypi ${{ secrets.PYPI_API_TOKEN }}
+          poetry publish --build

.gitignore

@@ -0,0 +1,2 @@
+.venv/
+dist/

pyproject.toml

@@ -1,25 +1,39 @@
-# This project can be installed with `python3 -m pip install -e .` from the main directory. 
+[tool.poetry]
-
-[project]
 name = "timesfm"
+version = "0.0.2"
 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",
-    "numpy>=1.26.4",
-    "pandas>=2.1.4",
-]
 authors = [
-     {name = "Rajat Sen", email = "senrajat@google.com"},
+    "Rajat Sen <senrajat@google.com>",
-     {name = "Yichen Zhou", email = "yichenzhou@google.com"},
+    "Yichen Zhou <yichenzhou@google.com>",
-     {name = "Abhimanyu Das", email = "abhidas@google.com"},
+    "Abhimanyu Das <abhidas@google.com>",
-     {name = "Petros Mol", email = "pmol@google.com"},
+    "Petros Mol <pmol@google.com>",
+    "Justin Güse <guese.justin@gmail.com>",
 ]
+readme = "README.md"
+keywords = ["time series", "timesfm", "forecast", "time series model"]
+homepage = "https://github.com/google-research/timesfm"
+repository = "https://github.com/google-research/timesfm"
+classifiers = [
+    "Environment :: Console",
+    "Framework :: Flake8",
+    "Operating System :: OS Independent",
+    "Topic :: Software Development :: Documentation",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "Topic :: Software Development :: Quality Assurance",
+]
+include = [
+    "LICENSE",
+]
+
+[tool.poetry.dependencies]
+python = ">=3.10,<3.11"
+einshape = ">=1.0.0"
+numpy = ">=1.26.4"
+pandas = ">=2.1.4"
+paxml = "1.4.0"
+jax = "0.4.26"
 
 
 [build-system]
-requires = ["setuptools"]
+requires = ["poetry-core"]
-build-backend = "setuptools.build_meta"
+build-backend = "poetry.core.masonry.api"