Expose new community endpoints under v2

Author: FlorentinD

graphdatascience/session/session_v2_endpoints.py

@@ -2,11 +2,16 @@
 
 from graphdatascience.arrow_client.authenticated_flight_client import AuthenticatedArrowClient
 from graphdatascience.arrow_client.v2.remote_write_back_client import RemoteWriteBackClient
+from graphdatascience.procedure_surface.api.community.clique_counting_endpoints import CliqueCountingEndpoints
+from graphdatascience.procedure_surface.api.community.kmeans_endpoints import KMeansEndpoints
+from graphdatascience.procedure_surface.api.community.labelpropagation_endpoints import LabelPropagationEndpoints
+from graphdatascience.procedure_surface.api.community.leiden_endpoints import LeidenEndpoints
 from graphdatascience.procedure_surface.arrow.articlerank_arrow_endpoints import ArticleRankArrowEndpoints
 from graphdatascience.procedure_surface.arrow.articulationpoints_arrow_endpoints import ArticulationPointsArrowEndpoints
 from graphdatascience.procedure_surface.arrow.betweenness_arrow_endpoints import BetweennessArrowEndpoints
 from graphdatascience.procedure_surface.arrow.catalog_arrow_endpoints import CatalogArrowEndpoints
 from graphdatascience.procedure_surface.arrow.celf_arrow_endpoints import CelfArrowEndpoints
+from graphdatascience.procedure_surface.arrow.clique_counting_arrow_endpoints import CliqueCountingArrowEndpoints
 from graphdatascience.procedure_surface.arrow.closeness_arrow_endpoints import ClosenessArrowEndpoints
 from graphdatascience.procedure_surface.arrow.closeness_harmonic_arrow_endpoints import ClosenessHarmonicArrowEndpoints
 from graphdatascience.procedure_surface.arrow.degree_arrow_endpoints import DegreeArrowEndpoints
@@ -17,6 +22,9 @@
 from graphdatascience.procedure_surface.arrow.hashgnn_arrow_endpoints import HashGNNArrowEndpoints
 from graphdatascience.procedure_surface.arrow.k1coloring_arrow_endpoints import K1ColoringArrowEndpoints
 from graphdatascience.procedure_surface.arrow.kcore_arrow_endpoints import KCoreArrowEndpoints
+from graphdatascience.procedure_surface.arrow.kmeans_arrow_endpoints import KMeansArrowEndpoints
+from graphdatascience.procedure_surface.arrow.labelpropagation_arrow_endpoints import LabelPropagationArrowEndpoints
+from graphdatascience.procedure_surface.arrow.leiden_arrow_endpoints import LeidenArrowEndpoints
 from graphdatascience.procedure_surface.arrow.louvain_arrow_endpoints import LouvainArrowEndpoints
 from graphdatascience.procedure_surface.arrow.node2vec_arrow_endpoints import Node2VecArrowEndpoints
 from graphdatascience.procedure_surface.arrow.pagerank_arrow_endpoints import PageRankArrowEndpoints
@@ -61,6 +69,12 @@ def articulation_points(self) -> ArticulationPointsArrowEndpoints:
     def betweenness_centrality(self) -> BetweennessArrowEndpoints:
         return BetweennessArrowEndpoints(self._arrow_client, self._write_back_client, show_progress=self._show_progress)
 
+    @property
+    def clique_counting(self) -> CliqueCountingEndpoints:
+        return CliqueCountingArrowEndpoints(
+            self._arrow_client, self._write_back_client, show_progress=self._show_progress
+        )
+
     @property
     def closeness_centrality(self) -> ClosenessArrowEndpoints:
         return ClosenessArrowEndpoints(self._arrow_client, self._write_back_client, show_progress=self._show_progress)
@@ -111,6 +125,19 @@ def k1_coloring(self) -> K1ColoringArrowEndpoints:
     def k_core_decomposition(self) -> KCoreArrowEndpoints:
         return KCoreArrowEndpoints(self._arrow_client, self._write_back_client, show_progress=self._show_progress)
 
+    @property
+    def kmeans(self) -> KMeansEndpoints:
+        return KMeansArrowEndpoints(self._arrow_client, self._write_back_client, show_progress=self._show_progress)
+
+    @property
+    def label_propagation(self) -> LabelPropagationEndpoints:
+        return LabelPropagationArrowEndpoints(
+            self._arrow_client, self._write_back_client, show_progress=self._show_progress
+        )
+
+    def leiden(self) -> LeidenEndpoints:
+        return LeidenArrowEndpoints(self._arrow_client, self._write_back_client, show_progress=self._show_progress)
+
     @property
     def louvain(self) -> LouvainArrowEndpoints:
         return LouvainArrowEndpoints(self._arrow_client, self._write_back_client, show_progress=self._show_progress)

graphdatascience/tests/integrationV2/procedure_surface/session/test_session_endpoint_coverage.py

@@ -9,24 +9,16 @@
 from graphdatascience.session.session_v2_endpoints import SessionV2Endpoints
 
 MISSING_ALGO_ENDPOINTS = {
-    "community.kmeans",
-    "community.cliquecounting",
     "community.maxkcut",
-    "community.cliquecounting.estimate",
-    "community.labelPropagation.estimate",
     "community.maxkcut.estimate",
     "community.k1coloring",
     "community.triangleCount.estimate",
-    "community.kmeans.estimate",
-    "community.leiden",
+    "community.sllpa",
     "community.sllpa.estimate",
     "community.modularityOptimization",
-    "community.sllpa",
     "community.localClusteringCoefficient",
-    "community.modularityOptimization.estimate",
-    "community.labelPropagation",
     "community.localClusteringCoefficient.estimate",
-    "community.leiden.estimate",
+    "community.modularityOptimization.estimate",
     "community.triangleCount",
     "embeddings.graphSage.train.estimate",  # TODO fix this by moving behind shared interface
     "embeddings.graphSage.estimate",
@@ -36,8 +28,8 @@
     "similarity.nodeSimilarity.filtered.estimate",
     "similarity.nodeSimilarity",
     "similarity.knn",
-    "similarity.nodeSimilarity.estimate",
     "similarity.knn.estimate",
+    "similarity.nodeSimilarity.estimate",
     "pathfinding.sourceTarget.dijkstra.estimate",
     "pathfinding.sourceTarget.aStar",
     "pathfinding.prizeSteinerTree.estimate",