From d9fc8c8e90045f0150298ac6fe27c57867f901bc Mon Sep 17 00:00:00 2001 From: Simon Adorf Date: Fri, 17 Oct 2025 17:11:20 -0500 Subject: [PATCH 1/3] Show installation path from pypi.org for some packges. --- _includes/selector.html | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/_includes/selector.html b/_includes/selector.html index f3223d7581c..c3657446a71 100644 --- a/_includes/selector.html +++ b/_includes/selector.html @@ -594,6 +594,12 @@ // Make sure all packages (and version selectors) are quoted pkgs = pkgs.flatMap(pkg => '"' + pkg + '"'); + // Check if only cuDF, cuML, and/or dask-cuDF are selected (exclude --extra-index-url for these) + var selected_packages = this.active_packages.filter(pkg => pkg !== "Choose Specific Packages"); + var pypi_available_packages = ["cudf", "cuml", "dask-cudf"]; + var only_pip_available_packages_selected = selected_packages.length > 0 && + selected_packages.every(pkg => pypi_available_packages.includes(pkg.toLowerCase())); + // For every n packages add a new line with a "\" character // We need i += n + 1 since the splice adds a new element to the array if (this.active_release === "Stable") { @@ -606,7 +612,7 @@ pkgs.splice(i, 0, "\\\n" + indentation.slice(0, -1)); } - return [pip_install, index_url, pkgs.flatMap(pkg => this.highlightPkgOrImg(pkg)).join(" ")] + return [pip_install, (only_pip_available_packages_selected ? null : index_url), pkgs.flatMap(pkg => this.highlightPkgOrImg(pkg)).join(" ")] .filter(Boolean) .join(" \\\n" + indentation) }, @@ -677,7 +683,17 @@ }, getpipNotes() { var notes = []; - var install_location_notes = "RAPIDS pip packages are hosted by NVIDIA
" + + // Check if only PyPI-available packages are selected + var selected_packages = this.active_packages.filter(pkg => pkg !== "Choose Specific Packages" && pkg !== "Standard"); + var pypi_available_packages = ["cudf", "cuml", "dask-cudf"]; + var only_pip_available_packages_selected = selected_packages.length > 0 && + selected_packages.every(pkg => pypi_available_packages.includes(pkg.toLowerCase())); + + var install_location_notes = only_pip_available_packages_selected + ? "RAPIDS pip packages are hosted by NVIDIA, some are also available on the Python Package Index (PyPI).
" + : "RAPIDS pip packages are hosted by NVIDIA
"; + var version_string = this.getSupportedPythonVersions().map((v) => `${v}`).join(", ") notes = [...notes, install_location_notes, `pip installation supports the following Python versions: ${version_string}.
`]; From be40e5a66f82435be8bb3ca337f6878d4c7a0144 Mon Sep 17 00:00:00 2001 From: Simon Adorf Date: Mon, 20 Oct 2025 09:46:14 -0500 Subject: [PATCH 2/3] Refactor selection logic. --- _includes/selector.html | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/_includes/selector.html b/_includes/selector.html index c3657446a71..f3428032185 100644 --- a/_includes/selector.html +++ b/_includes/selector.html @@ -536,6 +536,12 @@ removeLeadingZeros(version) { return version.split(".").map(Number).join("."); }, + areOnlyPypiAvailablePackagesSelected() { + var selected_packages = this.active_packages.filter(pkg => pkg !== "Choose Specific Packages" && pkg !== "Standard"); + var pypi_available_packages = ["cudf", "cuml", "dask-cudf"]; + return selected_packages.length > 0 && + selected_packages.every(pkg => pypi_available_packages.includes(pkg.toLowerCase())); + }, getpipCmdHtml() { var pip_install = `${this.highlightCmd("pip")} install`; var cuda_suffix = `-cu${this.active_pip_cuda_ver}`; @@ -594,11 +600,8 @@ // Make sure all packages (and version selectors) are quoted pkgs = pkgs.flatMap(pkg => '"' + pkg + '"'); - // Check if only cuDF, cuML, and/or dask-cuDF are selected (exclude --extra-index-url for these) - var selected_packages = this.active_packages.filter(pkg => pkg !== "Choose Specific Packages"); - var pypi_available_packages = ["cudf", "cuml", "dask-cudf"]; - var only_pip_available_packages_selected = selected_packages.length > 0 && - selected_packages.every(pkg => pypi_available_packages.includes(pkg.toLowerCase())); + // Check if only PyPI-available packages are selected (exclude --extra-index-url for these) + var only_pypi_available_packages_selected = this.areOnlyPypiAvailablePackagesSelected(); // For every n packages add a new line with a "\" character // We need i += n + 1 since the splice adds a new element to the array @@ -612,7 +615,7 @@ pkgs.splice(i, 0, "\\\n" + indentation.slice(0, -1)); } - return [pip_install, (only_pip_available_packages_selected ? null : index_url), pkgs.flatMap(pkg => this.highlightPkgOrImg(pkg)).join(" ")] + return [pip_install, (only_pypi_available_packages_selected ? null : index_url), pkgs.flatMap(pkg => this.highlightPkgOrImg(pkg)).join(" ")] .filter(Boolean) .join(" \\\n" + indentation) }, @@ -685,12 +688,9 @@ var notes = []; // Check if only PyPI-available packages are selected - var selected_packages = this.active_packages.filter(pkg => pkg !== "Choose Specific Packages" && pkg !== "Standard"); - var pypi_available_packages = ["cudf", "cuml", "dask-cudf"]; - var only_pip_available_packages_selected = selected_packages.length > 0 && - selected_packages.every(pkg => pypi_available_packages.includes(pkg.toLowerCase())); + var only_pypi_available_packages_selected = this.areOnlyPypiAvailablePackagesSelected(); - var install_location_notes = only_pip_available_packages_selected + var install_location_notes = only_pypi_available_packages_selected ? "RAPIDS pip packages are hosted by NVIDIA, some are also available on the Python Package Index (PyPI).
" : "RAPIDS pip packages are hosted by NVIDIA
"; From 44c378b57e74fe6cfcd57cade0c040290b4ea8e0 Mon Sep 17 00:00:00 2001 From: Simon Adorf Date: Mon, 20 Oct 2025 09:55:08 -0500 Subject: [PATCH 3/3] Add RAFT to PyPI installable packages. --- _includes/selector.html | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/_includes/selector.html b/_includes/selector.html index f3428032185..a960e02fa24 100644 --- a/_includes/selector.html +++ b/_includes/selector.html @@ -538,7 +538,7 @@ }, areOnlyPypiAvailablePackagesSelected() { var selected_packages = this.active_packages.filter(pkg => pkg !== "Choose Specific Packages" && pkg !== "Standard"); - var pypi_available_packages = ["cudf", "cuml", "dask-cudf"]; + var pypi_available_packages = ["cudf", "cuml", "dask-cudf", "raft"]; return selected_packages.length > 0 && selected_packages.every(pkg => pypi_available_packages.includes(pkg.toLowerCase())); },