Fix scikit-learn upperlimit for 23.0.0 release

openml · Jun 17, 2024 · d5e703d · d5e703d
1 parent 37a3d0d
commit d5e703d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/frameworks/GAMA/setup.sh b/frameworks/GAMA/setup.sh
@@ -23,4 +23,10 @@ else
     PIP install -U -e ${TARGET_DIR}
 fi
 
+if [[ "$VERSION" == "23.0.0" ]]; then
+    # We include this only because this is the fixed version for the 2023Q2 definition.
+    echo "GAMA/setup.sh: Downgrading scikit-learn to compatible version."
+    PIP install --no-cache-dir -U "scikit-learn<1.3"
+fi
+
 PY -c "from gama import __version__; print(__version__)" >> "${HERE}/.setup/installed"