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