Correct the python sssp result by setting generate_eid to false in lo…

…ad_p2p_network dataset (#2762) Fixes #2735
alibaba · May 29, 2023 · a119c23 · a119c23
1 parent e4bb069
commit a119c23
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 7 deletions.
diff --git a/docs/analytical_engine/tutorial_dev_algo_python.rst b/docs/analytical_engine/tutorial_dev_algo_python.rst
@@ -201,7 +201,7 @@ To run your own algorithms, you may trigger it in place where you defined it.
     import graphscope
     from graphscope.dataset import load_p2p_network
 
-    g = load_p2p_network()
+    g = load_p2p_network(generate_eid=False)
 
     # load my algorithm
     my_app = SSSP_Pregel()
@@ -227,4 +227,4 @@ Later, you can load your own algorithm from the gar package.
     my_app = load_app("/tmp/my_sssp_pregel.gar")
 
     # run my algorithm over a graph and get the result.
-    ret = my_app(g, src="6")
+    ret = my_app(g, src="6")
diff --git a/docs/zh/analytics_engine.rst b/docs/zh/analytics_engine.rst
@@ -319,7 +319,7 @@ GraphScope 支持用户在自定义算法中通过 :code:`context.math` 上的
     import graphscope
     from graphscope.dataset import load_p2p_network
 
-    g = load_p2p_network()
+    g = load_p2p_network(generate_eid=False)
 
     # 加载自己的算法
     my_app = SSSP_Pregel()

diff --git a/python/graphscope/dataset/p2p_network.py b/python/graphscope/dataset/p2p_network.py
@@ -23,7 +23,7 @@
 from graphscope.dataset.io_utils import download_file
 
 
-def load_p2p_network(sess=None, prefix=None, directed=False):
+def load_p2p_network(sess=None, prefix=None, directed=False, generate_eid=True):
     """Load p2p graph.
     A peer-to-peer dataset derived from Gnutella peer-to-peer network, August 31 2002,
     with generated data on vertices and edges. See more details here:
@@ -38,6 +38,8 @@ def load_p2p_network(sess=None, prefix=None, directed=False):
             source URL. Defaults to None.
         directed (bool, optional): Determine to load a directed or undirected graph.
             Defaults to True.
+        generate_eid (bool, optional): True will generate id for each edge as the first property.
+            Defaults to True.
 
     Returns:
         :class:`graphscope.framework.graph.GraphDAGNode`:
@@ -76,7 +78,7 @@ def load_p2p_network(sess=None, prefix=None, directed=False):
     if sess is None:
         sess = get_default_session()
 
-    graph = sess.g(directed=directed)
+    graph = sess.g(directed=directed, generate_eid=generate_eid)
     graph = graph.add_vertices(
         os.path.join(prefix, "p2p-31_property_v_0"), "host"
     ).add_edges(

diff --git a/tutorials/06_writing_your_own_algorithms.ipynb b/tutorials/06_writing_your_own_algorithms.ipynb
@@ -302,7 +302,7 @@
     "\n",
     "from graphscope.dataset import load_p2p_network\n",
     "\n",
-    "graph = load_p2p_network(directed=False)"
+    "graph = load_p2p_network(directed=False, generate_eid=False)"
    ]
   },
   {

diff --git a/tutorials/zh/06_writing_your_own_algorithms.ipynb b/tutorials/zh/06_writing_your_own_algorithms.ipynb
@@ -298,7 +298,7 @@
     "\n",
     "from graphscope.dataset import load_p2p_network\n",
     "\n",
-    "graph = load_p2p_network(directed=False)"
+    "graph = load_p2p_network(directed=False, generate_eid=False)"
    ]
   },
   {