diff --git a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt
index 8f5fb55..b9a9008 100644
--- a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt
+++ b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt
@@ -1,6 +1,7 @@
 joblib==1.3.2
 boto3==1.28.17
 pandas==1.5.3
-autogluon==0.8.2
+scikit-learn==1.3.0
+# autogluon==0.8.2
 pyarrow==13.0.0
 PyYAML==6.0.1