From 662d404778af4720f190caf98525e5e2d9aef2dd Mon Sep 17 00:00:00 2001 From: Ralph Liu Date: Fri, 18 Oct 2024 17:44:05 -0400 Subject: [PATCH] Add nx-cugraph as option under pip install selection --- _includes/selector.html | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/_includes/selector.html b/_includes/selector.html index 86d716ed7da..c153d823f7a 100644 --- a/_includes/selector.html +++ b/_includes/selector.html @@ -390,7 +390,7 @@ img_loc: ["NGC", "Dockerhub"], img_types: ["Base", "Notebooks"], packages: ["Standard", "Choose Specific Packages"], - additional_pip_packages: ["cuDF", "dask-cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"], + additional_pip_packages: ["cuDF", "dask-cuDF", "cuML", "cuGraph/nx-cugraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"], additional_rapids_packages: ["cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"], additional_packages: ["Graphistry", "JupyterLab", "NetworkX + nx-cugraph", "Plotly Dash", "PyTorch", "TensorFlow", "Xarray-Spatial"], note_prefix: "", @@ -549,6 +549,7 @@ if (pkg === "cuspatial/cuproj") return ["cuspatial" + cuda_suffix, "cuproj" + cuda_suffix]; if (pkg === "raft") return ["pylibraft" + cuda_suffix, "raft-dask" + cuda_suffix]; if (pkg === "cuvs") return ["cuvs" + cuda_suffix]; + if (pkg === "cugraph/nx-cugraph") return ["cugraph" + cuda_suffix, "nx-cugraph" + cuda_suffix]; return [pkg + cuda_suffix]; } } @@ -561,6 +562,7 @@ if (pkg === "cuspatial/cuproj") return ["\"cuspatial" + cuda_suffix, "\"cuproj" + cuda_suffix]; if (pkg === "raft") return ["\"pylibraft" + cuda_suffix, "\"raft-dask" + cuda_suffix]; if (pkg === "cuvs") return ["\"cuvs" + cuda_suffix, "\"pylibraft" + cuda_suffix]; + if (pkg === "cugraph/nx-cugraph") return ["cugraph" + cuda_suffix, "nx-cugraph" + cuda_suffix]; return ["\"" + pkg + cuda_suffix]; } }