Pin scikit-learn dependency for shap

Signed-off-by: Lee Yang <[email protected]>
NVIDIA · Jan 9, 2025 · e1dc638 · e1dc638
1 parent dccf8b8
commit e1dc638
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/user_tools/pyproject.toml b/user_tools/pyproject.toml
@@ -81,6 +81,7 @@ dependencies = [
     "xgboost==2.1.3",
     # used for model interpretability. python [3.9, 3.12]
     "shap==0.46.0",
+    "scikit-learn==1.5.2",
     # used for retrieving available memory on the host
     "psutil==6.1.1"
 ]