From 7dd13aab060e1f613f17f69bec404d90f4260cc1 Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Tue, 12 Sep 2023 14:49:02 +0100 Subject: [PATCH] add requiremnets --- .../pipeline/src/requirements/predictions/requirements-dev.txt | 2 +- .../src/pipeline/src/requirements/predictions/requirements.txt | 2 +- .../src/pipeline/src/requirements/training/requirements.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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