From dfe3a88131cd2be53d51afa5909199a2023f7925 Mon Sep 17 00:00:00 2001
From: galipremsagar <sagarprem75@gmail.com>
Date: Tue, 24 Sep 2024 18:26:35 +0000
Subject: [PATCH] retain

---
 ci/cudf_pandas_scripts/run_tests.sh | 4 ++--
 dependencies.yaml                   | 9 +++++++++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/ci/cudf_pandas_scripts/run_tests.sh b/ci/cudf_pandas_scripts/run_tests.sh
index 2570579fd72..c6228a4ef33 100755
--- a/ci/cudf_pandas_scripts/run_tests.sh
+++ b/ci/cudf_pandas_scripts/run_tests.sh
@@ -56,10 +56,10 @@ else
 
     echo "" > ./constraints.txt
     if [[ $RAPIDS_DEPENDENCIES == "oldest" ]]; then
-        # `py_test_cudf_pandas` constraints are for `[test]` not `[cudf-pandas-tests]`
+        # `test_python` constraints are for `[test]` not `[cudf-pandas-tests]`
         rapids-dependency-file-generator \
             --output requirements \
-            --file-key py_test_cudf_pandas \
+            --file-key test_python \
             --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION};dependencies=${RAPIDS_DEPENDENCIES}" \
         | tee ./constraints.txt
     fi
diff --git a/dependencies.yaml b/dependencies.yaml
index 3202bdda0f8..9e38ebe8b08 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -43,6 +43,15 @@ files:
     includes:
       - cuda_version
       - test_cpp
+  test_python:
+    output: none
+    includes:
+      - cuda_version
+      - py_version
+      - test_python_common
+      - test_python_cudf
+      - test_python_dask_cudf
+      - test_python_cudf_pandas
   test_python_cudf:
     output: none
     includes: