From 9d3a8b0216e0af157a2e995bca819a978d063ec4 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Thu, 29 Aug 2024 13:22:01 -0500
Subject: [PATCH] Exclude some packages, and collapse conda dry-run output.

---
 ci/check_conda_nightly_env.py | 16 ++++++++++++++++
 ci/test_conda_nightly_env.sh  | 12 ++++++++----
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/ci/check_conda_nightly_env.py b/ci/check_conda_nightly_env.py
index 9030a8a8..ddc5dc91 100644
--- a/ci/check_conda_nightly_env.py
+++ b/ci/check_conda_nightly_env.py
@@ -7,12 +7,28 @@
 
 OLD_PACKAGE_THRESHOLD_DAYS = 3
 
+EXCLUDED_PACKAGES = {
+    # These packages do not have date strings:
+    "cubinlinker",
+    "rapids-dask-dependency",
+    "libxgboost",
+    "py-xgboost",
+    "xgboost",
+    # TODO: Remove nx-cugraph after https://github.com/rapidsai/cugraph/pull/4639
+    "nx-cugraph",
+    # TODO: Do we want ucx-proc on rapidsai or from conda-forge?
+    "ucx-proc",
+}
+
 
 def is_rapids_nightly_package(package_info):
     return package_info["channel"] == "rapidsai-nightly"
 
 
 def get_package_date(package):
+    if package in EXCLUDED_PACKAGES:
+        return None
+
     # Matches 6 digits starting with "2", which should be YYMMDD
     date_re = r"_(2\d{5})_"
 
diff --git a/ci/test_conda_nightly_env.sh b/ci/test_conda_nightly_env.sh
index 5176fbf2..7fe46cee 100755
--- a/ci/test_conda_nightly_env.sh
+++ b/ci/test_conda_nightly_env.sh
@@ -8,11 +8,11 @@ CUDA_VERSION=${RAPIDS_CUDA_VERSION%.*}
 
 JSON_FILENAME="rapids_cuda${CUDA_VERSION}_py${RAPIDS_PY_VERSION}.json"
 
-echo "Creating conda environment with rapids=${RAPIDS_VERSION}, python=${RAPIDS_PY_VERSION}, cuda-version=${CUDA_VERSION}"
-#rapids-logger "Creating conda environment with rapids=${RAPIDS_VERSION}, python=${RAPIDS_PY_VERSION}, cuda-version=${CUDA_VERSION}"
+rapids-logger "Creating conda environment with rapids=${RAPIDS_VERSION}, python=${RAPIDS_PY_VERSION}, cuda-version=${CUDA_VERSION}"
 
-#rapids-conda-retry \
-conda \
+echo "::group::conda dry-run output"
+
+rapids-conda-retry \
     create \
     --solver=libmamba \
     -n rapids-${RAPIDS_VERSION} \
@@ -26,4 +26,8 @@ conda \
     --json \
     | tee "${JSON_FILENAME}"
 
+echo "::endgroup::"
+
+rapids-logger "Parsing results from conda dry-run with rapids=${RAPIDS_VERSION}, python=${RAPIDS_PY_VERSION}, cuda-version=${CUDA_VERSION}"
+
 python ci/check_conda_nightly_env.py "${JSON_FILENAME}"