diff --git a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements-dev.txt b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements-dev.txt
index 5aac406..b4679d0 100644
--- a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements-dev.txt
+++ b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements-dev.txt
@@ -1,7 +1,7 @@
 joblib==1.3.2
 boto3==1.28.17
 pandas==1.5.3
-scikit-learn==1.3.0
+autogluon==0.8.2
 pyarrow==13.0.0
 pre-commit==3.3.3
 sphinx==7.2.5
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 ab7c886..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,6 +1,6 @@
 joblib==1.3.2
 boto3==1.28.17
 pandas==1.5.3
-scikit-learn==1.3.0
+autogluon==0.8.2
 pyarrow==13.0.0
 PyYAML==6.0.1
diff --git a/modules/ml-pipeline/src/pipeline/src/requirements/training/requirements.txt b/modules/ml-pipeline/src/pipeline/src/requirements/training/requirements.txt
index 196dfe7..11f11a1 100644
--- a/modules/ml-pipeline/src/pipeline/src/requirements/training/requirements.txt
+++ b/modules/ml-pipeline/src/pipeline/src/requirements/training/requirements.txt
@@ -1,3 +1,3 @@
 boto3==1.28.41
 pandas==1.5.3
-scikit-learn==1.3.0
+autogluon==0.8.2