Skip to content

Commit 59612c3

Browse files
committed
Expose config endpoints in V2
1 parent 1538731 commit 59612c3

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

graphdatascience/procedure_surface/arrow/config_arrow_endpoints.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,23 +13,21 @@
1313

1414

1515
class ConfigArrowEndpoints(ConfigEndpoints):
16-
def __init__(self, arrow_client: AuthenticatedArrowClient, show_progress: bool = False):
16+
def __init__(self, arrow_client: AuthenticatedArrowClient):
1717
self._arrow_client = arrow_client
18-
self._show_progress = show_progress
1918

2019
@property
2120
def defaults(self) -> DefaultsEndpoints:
22-
return DefaultsArrowEndpoints(self._arrow_client, self._show_progress)
21+
return DefaultsArrowEndpoints(self._arrow_client)
2322

2423
@property
2524
def limits(self) -> LimitsEndpoints:
26-
return LimitsArrowEndpoints(self._arrow_client, self._show_progress)
25+
return LimitsArrowEndpoints(self._arrow_client)
2726

2827

2928
class DefaultsArrowEndpoints(DefaultsEndpoints):
30-
def __init__(self, arrow_client: AuthenticatedArrowClient, show_progress: bool = False):
29+
def __init__(self, arrow_client: AuthenticatedArrowClient):
3130
self._arrow_client = arrow_client
32-
self._show_progress = show_progress
3331

3432
def set(
3533
self,
@@ -58,9 +56,8 @@ def list(
5856

5957

6058
class LimitsArrowEndpoints(LimitsEndpoints):
61-
def __init__(self, arrow_client: AuthenticatedArrowClient, show_progress: bool = False):
59+
def __init__(self, arrow_client: AuthenticatedArrowClient):
6260
self._arrow_client = arrow_client
63-
self._show_progress = show_progress
6461

6562
def set(
6663
self,

graphdatascience/session/session_v2_endpoints.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@
5454
TriangleCountArrowEndpoints,
5555
)
5656
from graphdatascience.procedure_surface.arrow.community.wcc_arrow_endpoints import WccArrowEndpoints
57+
from graphdatascience.procedure_surface.arrow.config_arrow_endpoints import ConfigArrowEndpoints
5758
from graphdatascience.procedure_surface.arrow.node_embedding.fastrp_arrow_endpoints import FastRPArrowEndpoints
5859
from graphdatascience.procedure_surface.arrow.node_embedding.graphsage_predict_arrow_endpoints import (
5960
GraphSagePredictArrowEndpoints,
@@ -86,6 +87,10 @@ def set_show_progress(self, show_progress: bool) -> None:
8687
def graph(self) -> CatalogArrowEndpoints:
8788
return CatalogArrowEndpoints(self._arrow_client, self._db_client, show_progress=self._show_progress)
8889

90+
@property
91+
def config(self) -> ConfigArrowEndpoints:
92+
return ConfigArrowEndpoints(self._arrow_client)
93+
8994
## Algorithms
9095

9196
@property

0 commit comments

Comments
 (0)