move pins to ray

Nixtla · Aug 28, 2024 · f1891aa · f1891aa
1 parent 136f99e
commit f1891aa
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/.github/workflows/build-docs.yaml b/.github/workflows/build-docs.yaml
@@ -26,7 +26,7 @@ jobs:
       - name: Build docs
         run: |
           set -ux
-          pip install uv && uv pip install --system ".[all]"
+          pip install uv && uv pip install --system r -r pyproject.toml --all-extras
           uv pip install --system .
           mkdir nbs/_extensions
           cp -r docs-scripts/mintlify/ nbs/_extensions/

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -98,7 +98,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install pip requirements
-        run: pip install uv && uv pip install --system -r pyproject.toml --extra dev --extra plotly
+        run: pip install uv && uv pip install --system -v -r pyproject.toml --extra dev --extra plotly
 
       - name: Download wheels
         uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8
@@ -134,7 +134,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install pip requirements
-        run: pip install uv && uv pip install --system -r pyproject.toml --extra all
+        run: pip install uv && uv pip install --system -v -r pyproject.toml --all-extras
 
       - name: Download wheels
         uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8

diff --git a/pyproject.toml b/pyproject.toml
@@ -32,10 +32,6 @@ readme = "README.md"
 keywords = ["time-series", "forecasting", "arima", "ets"]
 
 [project.optional-dependencies]
-all = [
-    "statsforecast[dask,dev,plotly,polars,ray,spark] ; python_version < '3.12'",
-    "statsforecast[dask,dev,plotly,polars,spark] ; python_version >= '3.12'",
-]
 dask = ["fugue[dask]>=0.8.1"]
 dev = [
     "black",
@@ -55,7 +51,12 @@ dev = [
 ]
 plotly = ["nbdev_plotly", "plotly", "plotly-resampler"]
 polars = ["polars[numpy]"]
-ray = ["fugue[ray]>=0.8.1", "protobuf>=3.15.3,<4", "numpy<2", "ray<2.8"]
+ray = [
+    "fugue[ray]>=0.8.1 ; python_version < '3.12'",
+    "protobuf>=3.15.3,<4 ; python_version < '3.12'",
+    "numpy<2 ; python_version < '3.12'",
+    "ray<2.8 ; python_version < '3.12'",
+]
 spark = ["fugue[spark]>=0.8.1"]
 
 [project.entry-points.nbdev]