From 242f7a6c85d7be5f72af7956e03741e63145e92f Mon Sep 17 00:00:00 2001
From: Ralph Liu <137829296+nv-rliu@users.noreply.github.com>
Date: Mon, 23 Sep 2024 13:54:07 -0400
Subject: [PATCH] Adding `nx-cugraph` to RAPIDS Install Selector Tool (#536)

Co-authored-by: Bradley Dice <bdice@bradleydice.com>
---
 _includes/selector.html | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

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: "<i class='fas fa-info-circle text-blue'></i>",
-            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));