Generate pinning.

rapidsai · Jul 22, 2024 · 8472be0 · 8472be0
1 parent c0fc8ec
commit 8472be0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/_includes/selector.html b/_includes/selector.html
@@ -447,8 +447,9 @@
                 var rapids_channel = this.active_release === "Stable" ? "rapidsai" : "rapidsai-nightly";
                 var dask_prefix = this.active_release === "Nightly" ? "dask/label/dev::" : "";
                 var python_version = this.active_python_ver;
-                var cuda_version = this.active_conda_cuda_ver;
-                var py_cuda_pkgs = [this.highlightPkgOrImg("python") + "=" + python_version, this.highlightPkgOrImg("cuda-version") + "=" + cuda_version].join(" ");
+                var cuda_version_bounds = this.active_conda_cuda_ver.split(" - ");
+                var cuda_version_pinning = ">=" + cuda_version_bounds[0] + ",<=" + cuda_version_bounds[1];
+                var py_cuda_pkgs = [this.highlightPkgOrImg("python") + "=" + python_version, this.highlightPkgOrImg("cuda-version") + cuda_version_pinning].join(" ");
                 var conda_channels = [rapids_channel, "conda-forge", "nvidia"]
                     .map(ch => "-" + this.highlightFlag("c") + " " + ch + " ")
                     .join("");