Exclude some packages, and collapse conda dry-run output.

rapidsai · Aug 29, 2024 · 9d3a8b0 · 9d3a8b0
1 parent 55da82b
commit 9d3a8b0
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git 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
@@ -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}"