diff --git a/_includes/selector.html b/_includes/selector.html
index 906b1012238..9c9932d3563 100644
--- a/_includes/selector.html
+++ b/_includes/selector.html
@@ -392,9 +392,9 @@
packages: ["Standard", "Choose Specific Packages"],
additional_pip_packages: ["cuDF", "dask-cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"],
additional_rapids_packages: ["cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"],
- additional_packages: ["JupyterLab", "Plotly Dash", "Graphistry", "TensorFlow", "Xarray-Spatial", "PyTorch"],
+ additional_packages: ["Graphistry", "JupyterLab", "NetworkX + nx-cugraph", "Plotly Dash", "PyTorch", "TensorFlow", "Xarray-Spatial"],
note_prefix: "",
- rapids_meta_pkgs: ["cuDF", "cuML", "cuGraph", "cuSpatial", "cuProj", "cuxfilter", "cuCIM", "RAFT"],
+ rapids_meta_pkgs: ["cuDF", "cuML", "cuGraph", "nx-cugraph", "cuSpatial", "cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"],
getStableVersion() {
return "{{ site.data.releases.stable.version }}";
},
@@ -456,7 +456,10 @@
};
},
getAdditionalPkgName(pkg) {
+ // used to pin the version of nx-cugraph for conda install
+ var rapids_version = this.active_release === "Stable" ? this.getStableVersion() : this.getNightlyVersion();
var pkg_names = {
+ "NetworkX + nx-cugraph": `networkx nx-cugraph=${rapids_version}`,
"Plotly Dash": "dash",
"PyTorch": "'pytorch=*=*cuda*'"
}
@@ -539,7 +542,7 @@
// This has duplicate code, but makes for easier edits in the future
if (this.active_release === "Stable") {
index_url = `--${this.highlightFlag("extra-index-url")}=https://pypi.nvidia.com`;
- var version = this.removeLeadingZeros("{{ site.data.releases.stable.version }}")
+ var version = this.removeLeadingZeros("{{ site.data.releases.stable.version }}");
cuda_suffix = cuda_suffix + `==${version}.*`;
var libraryToPkg = (pkg) => {
pkg = pkg.toLowerCase();
@@ -566,6 +569,8 @@
return "Select at least one package.";
} else if (this.active_packages[0] === 'Standard') {
var pkgs = this.additional_pip_packages.flatMap(libraryToPkg);
+ var additional_pip_commands = libraryToPkg("nx-cugraph");
+ pkgs = pkgs.concat(additional_pip_commands);
} else {
// sort active_packages to appear in the same order as the additional_pip_packages list
this.active_packages.sort((a, b) => this.additional_pip_packages.indexOf(a) - this.additional_pip_packages.indexOf(b));