Pin scikit-learn dependency for shap (#1491)

Pin scikit-learn (dependency of shap) to version 1.5.2 to fix optuna runtime errors. --------- Signed-off-by: Lee Yang <[email protected]>
NVIDIA · Jan 9, 2025 · a215df7 · a215df7
1 parent dccf8b8
commit a215df7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/user_tools/pyproject.toml b/user_tools/pyproject.toml
@@ -81,6 +81,8 @@ dependencies = [
     "xgboost==2.1.3",
     # used for model interpretability. python [3.9, 3.12]
     "shap==0.46.0",
+    # dependency of shap, python [3.9, 3.12]
+    "scikit-learn==1.5.2",
     # used for retrieving available memory on the host
     "psutil==6.1.1"
 ]