updating workflows

2025-10-08 00:06:52 +00:00
parent 39df49c3a0
commit 1c0cd87f2b
2 changed files with 3 additions and 3 deletions
@@ -16,9 +16,9 @@ jobs:
        run: |
          curl -LsSf https://astral.sh/uv/install.sh | sh
          echo "$HOME/.cargo/bin" >> $GITHUB_PATH
-      - name: Install Poetry
+      - name: Install Poetry and dependencies
        run: |
-          uv pip install poetry
+          uv pip install poetry ".[torch,flax]"
      - name: Build and Publish to PyPI
        run: |
          poetry config pypi-token.pypi ${{ secrets.PYPI_API_TOKEN }}