From d720ae4dde786e7617bd5d2832d8de2ae92da23c Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Fri, 15 Sep 2023 00:27:07 +0100 Subject: [PATCH] just to make builds quicker as we will showcase sklearn models --- .../src/pipeline/src/requirements/predictions/requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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