ci: install pyarrow from nightly channel in a separate command

Author: kszucs

.github/workflows/ci.yml

@@ -25,7 +25,8 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple .[quality]
+          pip install --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple "pyarrow>=21.0.0.dev"
+          pip install .[quality]
       - name: Check quality
         run: |
           ruff check tests src benchmarks utils setup.py # linter
@@ -56,7 +57,9 @@ jobs:
       - name: Install uv
         run: pip install --upgrade uv
       - name: Install dependencies
-        run: uv pip install --system --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple "datasets[tests] @ ."
+        run: |
+          uv pip install --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple "pyarrow>=21.0.0.dev"
+          uv pip install --system "datasets[tests] @ ."
       - name: Install dependencies (latest versions)
         if: ${{ matrix.deps_versions == 'deps-latest' }}
         run: uv pip install --system --upgrade huggingface-hub "dill<0.3.9"
@@ -89,7 +92,9 @@ jobs:
       - name: Install uv
         run: pip install --upgrade uv
       - name: Install dependencies
-        run: uv pip install --system --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple "datasets[tests] @ ."
+        run: |
+          uv pip install --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple "pyarrow>=21.0.0.dev"
+          uv pip install --system "datasets[tests] @ ."
       - name: Test with pytest
         run: |
           python -m pytest -rfExX -m ${{ matrix.test }} -n 2 --dist loadfile -sv ./tests/
@@ -116,7 +121,9 @@ jobs:
       - name: Install uv
         run: pip install --upgrade uv
       - name: Install dependencies
-        run: uv pip install --system --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple "datasets[tests_numpy2] @ ."
+        run: |
+          uv pip install --extra-index-url https://pypi.anaconda.org/scientific-python-nightly-wheels/simple "pyarrow>=21.0.0.dev"
+          uv pip install --system "datasets[tests_numpy2] @ ."
       - name: Test with pytest
         run: |
           python -m pytest -rfExX -m ${{ matrix.test }} -n 2 --dist loadfile -sv ./tests/