test pinning metapackage generated by conda-lock

.github/workflows/pr.yaml

@@ -25,11 +25,11 @@ jobs:
       - conda-notebook-tests
       - docs-build
       - telemetry-setup
-      - wheel-build-libcuml
-      - wheel-build-cuml
-      - wheel-tests-cuml
-      - wheel-tests-cuml-dask
-      - devcontainer
+      # - wheel-build-libcuml
+      # - wheel-build-cuml
+      # - wheel-tests-cuml
+      # - wheel-tests-cuml-dask
+      # - devcontainer
     secrets: inherit
     uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@branch-25.10
     if: always()
@@ -215,60 +215,60 @@ jobs:
       arch: "amd64"
       container_image: "rapidsai/ci-conda:25.10-latest"
       script: "ci/build_docs.sh"
-  wheel-build-libcuml:
-    needs: checks
-    secrets: inherit
-    uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@branch-25.10
-    with:
-      build_type: pull-request
-      branch: ${{ inputs.branch }}
-      sha: ${{ inputs.sha }}
-      date: ${{ inputs.date }}
-      script: ci/build_wheel_libcuml.sh
-      extra-repo: rapidsai/cumlprims_mg
-      extra-repo-sha: branch-25.10
-      extra-repo-deploy-key: CUMLPRIMS_SSH_PRIVATE_DEPLOY_KEY
-      # build for every combination of arch and CUDA version, but only for the latest Python
-      matrix_filter: group_by([.ARCH, (.CUDA_VER|split(".")|map(tonumber)|.[0])]) | map(max_by(.PY_VER|split(".")|map(tonumber)))
-      package-name: libcuml
-      package-type: cpp
-  wheel-build-cuml:
-    needs: [checks, wheel-build-libcuml]
-    secrets: inherit
-    uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@branch-25.10
-    with:
-      build_type: pull-request
-      script: ci/build_wheel_cuml.sh
-      package-name: cuml
-      package-type: python
-  wheel-tests-cuml:
-    needs: [wheel-build-cuml, changed-files]
-    secrets: inherit
-    uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@branch-25.10
-    if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python
-    with:
-      build_type: pull-request
-      script: ci/test_wheel.sh
-  wheel-tests-cuml-dask:
-    needs: [wheel-build-cuml, changed-files]
-    secrets: inherit
-    uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@branch-25.10
-    if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python
-    with:
-      build_type: pull-request
-      script: ci/test_wheel_dask.sh
-  devcontainer:
-    needs: telemetry-setup
-    secrets: inherit
-    uses: rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@branch-25.10
-    with:
-      arch: '["amd64"]'
-      cuda: '["12.9"]'
-      extra-repo-deploy-key: CUMLPRIMS_SSH_PRIVATE_DEPLOY_KEY
-      build_command: |
-        sccache -z;
-        build-all --verbose -DBUILD_TESTS=ON -DBUILD_BENCHMARKS=ON;
-        sccache -s;
+  # wheel-build-libcuml:
+  #   needs: checks
+  #   secrets: inherit
+  #   uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@branch-25.10
+  #   with:
+  #     build_type: pull-request
+  #     branch: ${{ inputs.branch }}
+  #     sha: ${{ inputs.sha }}
+  #     date: ${{ inputs.date }}
+  #     script: ci/build_wheel_libcuml.sh
+  #     extra-repo: rapidsai/cumlprims_mg
+  #     extra-repo-sha: branch-25.10
+  #     extra-repo-deploy-key: CUMLPRIMS_SSH_PRIVATE_DEPLOY_KEY
+  #     # build for every combination of arch and CUDA version, but only for the latest Python
+  #     matrix_filter: group_by([.ARCH, (.CUDA_VER|split(".")|map(tonumber)|.[0])]) | map(max_by(.PY_VER|split(".")|map(tonumber)))
+  #     package-name: libcuml
+  #     package-type: cpp
+  # wheel-build-cuml:
+  #   needs: [checks, wheel-build-libcuml]
+  #   secrets: inherit
+  #   uses: rapidsai/shared-workflows/.github/workflows/wheels-build.yaml@branch-25.10
+  #   with:
+  #     build_type: pull-request
+  #     script: ci/build_wheel_cuml.sh
+  #     package-name: cuml
+  #     package-type: python
+  # wheel-tests-cuml:
+  #   needs: [wheel-build-cuml, changed-files]
+  #   secrets: inherit
+  #   uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@branch-25.10
+  #   if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python
+  #   with:
+  #     build_type: pull-request
+  #     script: ci/test_wheel.sh
+  # wheel-tests-cuml-dask:
+  #   needs: [wheel-build-cuml, changed-files]
+  #   secrets: inherit
+  #   uses: rapidsai/shared-workflows/.github/workflows/wheels-test.yaml@branch-25.10
+  #   if: fromJSON(needs.changed-files.outputs.changed_file_groups).test_python
+  #   with:
+  #     build_type: pull-request
+  #     script: ci/test_wheel_dask.sh
+  # devcontainer:
+  #   needs: telemetry-setup
+  #   secrets: inherit
+  #   uses: rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@branch-25.10
+  #   with:
+  #     arch: '["amd64"]'
+  #     cuda: '["12.9"]'
+  #     extra-repo-deploy-key: CUMLPRIMS_SSH_PRIVATE_DEPLOY_KEY
+  #     build_command: |
+  #       sccache -z;
+  #       build-all --verbose -DBUILD_TESTS=ON -DBUILD_BENCHMARKS=ON;
+  #       sccache -s;
   telemetry-summarize:
     # This job must use a self-hosted runner to record telemetry traces.
     runs-on: linux-amd64-cpu4

ci/build_cpp.sh

@@ -20,6 +20,7 @@ export RAPIDS_PACKAGE_VERSION
 
 # populates `RATTLER_CHANNELS` array and `RATTLER_ARGS` array
 source rapids-rattler-channel-string
+RATTLER_CHANNELS+=("-c" "msarahan")
 
 # --no-build-id allows for caching with `sccache`
 # more info is available at

ci/build_python.sh

@@ -26,6 +26,7 @@ source rapids-rattler-channel-string
 rapids-logger "Prepending channel ${CPP_CHANNEL} to RATTLER_CHANNELS"
 
 RATTLER_CHANNELS=("--channel" "${CPP_CHANNEL}" "${RATTLER_CHANNELS[@]}")
+RATTLER_CHANNELS+=("-c" "msarahan")
 
 sccache --zero-stats
 

conda/environments/all_cuda-129_arch-aarch64.yaml

@@ -4,6 +4,7 @@ channels:
 - rapidsai
 - rapidsai-nightly
 - conda-forge
+- msarahan
 dependencies:
 - c-compiler
 - certifi
@@ -61,6 +62,7 @@ dependencies:
 - rapids-build-backend>=0.3.0,<0.4.0.dev0
 - rapids-dask-dependency==25.10.*,>=0.0.0a0
 - rapids-logger==0.1.*,>=0.0.0a0
+- rapids-pin
 - recommonmark
 - rich
 - rmm==25.10.*,>=0.0.0a0

conda/environments/all_cuda-129_arch-x86_64.yaml

@@ -4,6 +4,7 @@ channels:
 - rapidsai
 - rapidsai-nightly
 - conda-forge
+- msarahan
 dependencies:
 - c-compiler
 - certifi
@@ -61,6 +62,7 @@ dependencies:
 - rapids-build-backend>=0.3.0,<0.4.0.dev0
 - rapids-dask-dependency==25.10.*,>=0.0.0a0
 - rapids-logger==0.1.*,>=0.0.0a0
+- rapids-pin
 - recommonmark
 - rich
 - rmm==25.10.*,>=0.0.0a0

conda/environments/clang_tidy_cuda-129_arch-x86_64.yaml

@@ -4,6 +4,7 @@ channels:
 - rapidsai
 - rapidsai-nightly
 - conda-forge
+- msarahan
 dependencies:
 - c-compiler
 - clang-tools==15.0.7

conda/environments/cpp_all_cuda-129_arch-x86_64.yaml

@@ -4,6 +4,7 @@ channels:
 - rapidsai
 - rapidsai-nightly
 - conda-forge
+- msarahan
 dependencies:
 - c-compiler
 - cmake>=3.30.4

conda/recipes/cuml/recipe.yaml

@@ -10,6 +10,8 @@ context:
   py_version: ${{ env.get("RAPIDS_PY_VERSION") }}
   py_buildstring: ${{ py_version | version_to_buildstring }}
   head_rev: '${{ git.head_rev(".")[:8] }}'
+  pin_version: '${{ minor_version }}.*'
+  pin_build_wildcard: '*_0'
 
 package:
   name: cuml
@@ -55,6 +57,7 @@ requirements:
     - ${{ compiler("cuda") }}
     - cuda-version =${{ cuda_version }}
     - ${{ stdlib("c") }}
+    - cuml-rapids-pins-build ${{ pin_version }} ${{ pin_build_wildcard }}
   host:
     - cuda-version =${{ cuda_version }}
     - cudf =${{ minor_version }}
@@ -71,6 +74,7 @@ requirements:
     - treelite ${{ treelite_version }}
     - cuda-cudart-dev
     - cuda-python >=12.6.2,<13.0a0
+    - cuml-rapids-pins-host ${{ pin_version }} ${{ pin_build_wildcard }}
   run:
     - ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
     - cudf =${{ minor_version }}

conda/recipes/libcuml/recipe.yaml

@@ -8,6 +8,8 @@ context:
   cuda_major: '${{ (env.get("RAPIDS_CUDA_VERSION") | split("."))[0] }}'
   date_string: '${{ env.get("RAPIDS_DATE_STRING") }}'
   head_rev: '${{ git.head_rev(".")[:8] }}'
+  pin_version: '${{ minor_version }}.*'
+  pin_build_wildcard: '*_0'
 
 recipe:
   name: libcuml-split
@@ -54,6 +56,7 @@ cache:
       - cmake ${{ cmake_version }}
       - ninja
       - ${{ stdlib("c") }}
+      - cuml-rapids-pins-build ${{ pin_version }} ${{ pin_build_wildcard }}
     host:
       - cuda-version =${{ cuda_version }}
       - libcumlprims =${{ minor_version }}
@@ -68,6 +71,7 @@ cache:
       - libcurand-dev
       - libcusolver-dev
       - libcusparse-dev
+      - cuml-rapids-pins-host ${{ pin_version }} ${{ pin_build_wildcard }}
 
 outputs:
   - package:
@@ -89,6 +93,7 @@ outputs:
       build:
         - cmake ${{ cmake_version }}
         - ${{ stdlib("c") }}
+        - cuml-rapids-pins-build ${{ pin_version }} ${{ pin_build_wildcard }}
       host:
         - cuda-version =${{ cuda_version }}
         - rapids-logger =0.1
@@ -102,6 +107,7 @@ outputs:
         - libcurand
         - libcusolver
         - libcusparse
+        - cuml-rapids-pins-host ${{ pin_version }} ${{ pin_build_wildcard }}
       run:
         - ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
         - cuda-cudart
@@ -115,6 +121,7 @@ outputs:
         - librmm =${{ minor_version }}
         - rapids-logger =0.1
         - treelite ${{ treelite_version }}
+
       ignore_run_exports:
         by_name:
           - cuda-cudart
@@ -143,11 +150,13 @@ outputs:
       build:
         - cmake ${{ cmake_version }}
         - ${{ stdlib("c") }}
+        - cuml-rapids-pins-build ${{ pin_version }} ${{ pin_build_wildcard }}
       host:
         - cuda-version =${{ cuda_version }}
         - rapids-logger =0.1
         - ${{ pin_subpackage("libcuml", exact=True) }}
         - cuda-cudart-dev
+        - cuml-rapids-pins-host ${{ pin_version }} ${{ pin_build_wildcard }}
       run:
         - ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
         - ${{ pin_subpackage("libcuml", exact=True) }}

dependencies.yaml

@@ -29,6 +29,7 @@ files:
       - rapids_build_backend
       - test_python
       - test_python_xgboost
+      - rapids_conda_pinning
   devcontainers:
     output: none
     includes:
@@ -55,6 +56,7 @@ files:
       - py_version
       - rapids_build_backend
       - test_python
+      - rapids_conda_pinning
   cpp_all:
     output: conda
     matrix:
@@ -208,6 +210,7 @@ channels:
   - rapidsai
   - rapidsai-nightly
   - conda-forge
+  - msarahan
 dependencies:
   rapids_build_backend:
     common:
@@ -295,6 +298,11 @@ dependencies:
           # pip recognizes the index as a global option for the requirements.txt file
           - --extra-index-url=https://pypi.nvidia.com
           - --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple
+  rapids_conda_pinning:
+    common:
+      - output_types: [conda]
+        packages:
+          - rapids-pin
   cuda_version:
     specific:
       - output_types: conda