Keep gds.alpha.graph.sample.rwr endpoint

Author: orazve

graphdatascience/graph/graph_alpha_proc_runner.py

@@ -11,14 +11,14 @@
 from .graph_alpha_project_runner import GraphAlphaProjectRunner
 from .graph_entity_ops_runner import GraphLabelRunner, GraphPropertyRunner
 from .graph_object import Graph
-from .graph_sample_runner import GraphSampleRunner
+from .graph_sample_runner import GraphAlphaSampleRunner
 
 
 class GraphAlphaProcRunner(UncallableNamespace, IllegalAttrChecker):
     @property
-    def sample(self) -> GraphSampleRunner:
+    def sample(self) -> GraphAlphaSampleRunner:
         self._namespace += ".sample"
-        return GraphSampleRunner(self._query_runner, self._namespace, self._server_version)
+        return GraphAlphaSampleRunner(self._query_runner, self._namespace, self._server_version)
 
     @property
     def graphProperty(self) -> GraphPropertyRunner:

graphdatascience/graph/graph_sample_runner.py

@@ -2,13 +2,23 @@
 
 from pandas import Series
 
+from ..error.deprecation_warning import deprecation_warning
 from ..error.illegal_attr_checker import IllegalAttrChecker
 from ..server_version.compatible_with import compatible_with
 from ..server_version.server_version import ServerVersion
 from .graph_object import Graph
 from .graph_type_check import from_graph_type_check
 
 
+class GraphAlphaSampleRunner(IllegalAttrChecker):
+    @compatible_with("construct", min_inclusive=ServerVersion(2, 2, 0))
+    @deprecation_warning("gds.graph.sample.rwr", ServerVersion(2, 4, 0))
+    @from_graph_type_check
+    def rwr(self, graph_name: str, from_G: Graph, **config: Any) -> Tuple[Graph, "Series[Any]"]:
+        runner = RWRRunner(self._query_runner, self._namespace + ".rwr", self._server_version)
+        return runner(graph_name, from_G, **config)
+
+
 class GraphSampleRunner(IllegalAttrChecker):
     @property
     def rwr(self) -> "RWRRunner":

graphdatascience/tests/unit/resources/example_server_endpoints.py

@@ -27,6 +27,7 @@
     "gds.alpha.graph.graphProperty.stream",
     "gds.alpha.graph.nodeLabel.mutate",
     "gds.alpha.graph.nodeLabel.write",
+    "gds.alpha.graph.sample.rwr",
     "gds.alpha.hits.mutate",
     "gds.alpha.hits.mutate.estimate",
     "gds.alpha.hits.stats",

graphdatascience/tests/unit/test_graph_ops.py

@@ -602,6 +602,18 @@ def test_graph_generate(runner: CollectingQueryRunner, gds: GraphDataScience) ->
     }
 
 
+def test_alpha_graph_sample_rwr(runner: CollectingQueryRunner, gds: GraphDataScience) -> None:
+    from_G, _ = gds.graph.project("g", "*", "*")
+    gds.alpha.graph.sample.rwr("s", from_G, samplingRatio=0.9, concurrency=7)
+
+    assert runner.last_query() == "CALL gds.alpha.graph.sample.rwr($graph_name, $from_graph_name, $config)"
+    assert runner.last_params() == {
+        "graph_name": "s",
+        "from_graph_name": "g",
+        "config": {"samplingRatio": 0.9, "concurrency": 7},
+    }
+
+
 def test_graph_sample_rwr(runner: CollectingQueryRunner, gds: GraphDataScience) -> None:
     from_G, _ = gds.graph.project("g", "*", "*")
     gds.graph.sample.rwr("s", from_G, samplingRatio=0.9, concurrency=7)