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