upper bound on sklearn for distributed

Nixtla · Dec 16, 2024 · 0d3ef60 · 0d3ef60
1 parent 1013383
commit 0d3ef60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -32,7 +32,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install the library
-        run: pip install uv && uv pip install --system ".[all]" "numba>=0.60" shap window-ops
+        run: pip install uv && uv pip install --system ".[all]" "numba>=0.60" "scikit-learn<1.6" shap window-ops
 
       - name: Run all tests
         run: nbdev_test --n_workers 0 --do_print --timing --skip_file_re 'electricity' --flags 'polars shap window_ops'