From 86fb71012dc4c166907e005ef4346f4de696c6c1 Mon Sep 17 00:00:00 2001 From: Mhamad Chamseddin Date: Thu, 7 May 2026 13:54:16 -0600 Subject: [PATCH] fix prefetch --- .tekton/odh-automl-pull-request.yaml | 3 ++- .../requirements-extra.txt | 7 +++++++ .../autogluon_tabular_training_pipeline/requirements.txt | 6 ------ 3 files changed, 9 insertions(+), 7 deletions(-) create mode 100644 pipelines/training/automl/autogluon_tabular_training_pipeline/requirements-extra.txt diff --git a/.tekton/odh-automl-pull-request.yaml b/.tekton/odh-automl-pull-request.yaml index e32979e44..6f83a5ca0 100644 --- a/.tekton/odh-automl-pull-request.yaml +++ b/.tekton/odh-automl-pull-request.yaml @@ -56,7 +56,8 @@ spec: "type": "pip", "path": "pipelines/training/automl", "requirements_files": [ - "autogluon_tabular_training_pipeline/requirements.txt" + "autogluon_tabular_training_pipeline/requirements.txt", + "autogluon_tabular_training_pipeline/requirements-extra.txt" ], "binary": {"arch": ":all:"} } diff --git a/pipelines/training/automl/autogluon_tabular_training_pipeline/requirements-extra.txt b/pipelines/training/automl/autogluon_tabular_training_pipeline/requirements-extra.txt new file mode 100644 index 000000000..62eaeda1b --- /dev/null +++ b/pipelines/training/automl/autogluon_tabular_training_pipeline/requirements-extra.txt @@ -0,0 +1,7 @@ +--index-url https://console.redhat.com/api/pypi/public-rhai/rhoai/3.4/cpu-ubi9-test/simple + +autogluon.common==1.5.0+rhaiv.2 +autogluon.core==1.5.0+rhaiv.2 +autogluon.features==1.5.0+rhaiv.2 +autogluon.tabular==1.5.0+rhaiv.2 +autogluon.timeseries==1.5.0+rhaiv.2 diff --git a/pipelines/training/automl/autogluon_tabular_training_pipeline/requirements.txt b/pipelines/training/automl/autogluon_tabular_training_pipeline/requirements.txt index f7e5f26e3..5f86d8e12 100644 --- a/pipelines/training/automl/autogluon_tabular_training_pipeline/requirements.txt +++ b/pipelines/training/automl/autogluon_tabular_training_pipeline/requirements.txt @@ -1,5 +1,4 @@ --index-url https://console.redhat.com/api/pypi/public-rhai/rhoai/3.4/cpu-ubi9/simple ---extra-index-url https://console.redhat.com/api/pypi/public-rhai/rhoai/3.4/cpu-ubi9-test/simple absl-py==2.4.0 accelerate==1.13.0 @@ -14,11 +13,6 @@ anyio==4.12.1 apsw==3.51.3.0 apswutils==0.1.2 attrs==26.1.0 -autogluon.common==1.5.0+rhaiv.2 -autogluon.core==1.5.0+rhaiv.2 -autogluon.features==1.5.0+rhaiv.2 -autogluon.tabular==1.5.0+rhaiv.2 -autogluon.timeseries==1.5.0+rhaiv.2 beartype==0.22.9 beautifulsoup4==4.14.3 blis==1.3.3