From 238f9096b422ce5dd4b0aa57a4efc64950313308 Mon Sep 17 00:00:00 2001
From: Brad Rees <34135411+BradReesWork@users.noreply.github.com>
Date: Fri, 23 Sep 2022 13:45:52 -0400
Subject: [PATCH] fix Comms import (#2717)

closes #2563

Authors:
  - Brad Rees (https://github.com/BradReesWork)

Approvers:
  - Don Acosta (https://github.com/acostadon)
  - Rick Ratzel (https://github.com/rlratzel)

URL: https://github.com/rapidsai/cugraph/pull/2717
---
 notebooks/demo/mg_pagerank.ipynb | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/notebooks/demo/mg_pagerank.ipynb b/notebooks/demo/mg_pagerank.ipynb
index db1f69ccbd2..7c32c052017 100644
--- a/notebooks/demo/mg_pagerank.ipynb
+++ b/notebooks/demo/mg_pagerank.ipynb
@@ -43,7 +43,7 @@
     "# Import needed libraries. We recommend using cugraph_dev env through conda\n",
     "from dask.distributed import Client, wait\n",
     "from dask_cuda import LocalCUDACluster\n",
-    "import cugraph.comms as Comms\n",
+    "from cugraph.dask.comms import comms as Comms\n",
     "import cugraph.dask as dask_cugraph\n",
     "import cugraph\n",
     "import dask_cudf\n",
@@ -162,7 +162,7 @@
    "outputs": [],
    "source": [
     "# Create a directed graph using the source (src) and destination (dst) vertex pairs from the Dataframe \n",
-    "G = cugraph.DiGraph()\n",
+    "G = cugraph.Graph(directed=True)\n",
     "G.from_dask_cudf_edgelist(e_list, source='src', destination='dst')\n",
     "\n",
     "# Print time\n",
@@ -295,9 +295,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "cugraph_dev",
    "language": "python",
-   "name": "python3"
+   "name": "cugraph_dev"
   },
   "language_info": {
    "codemirror_mode": {
@@ -309,7 +309,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.6"
+   "version": "3.9.13"
   }
  },
  "nbformat": 4,