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));