Skip to content

Commit 0737dd9

Browse files
committed
Expose config endpoints in V2
1 parent 1bf8e7a commit 0737dd9

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
@@ -86,6 +86,7 @@
8686
TriangleCountArrowEndpoints,
8787
)
8888
from graphdatascience.procedure_surface.arrow.community.wcc_arrow_endpoints import WccArrowEndpoints
89+
from graphdatascience.procedure_surface.arrow.config_arrow_endpoints import ConfigArrowEndpoints
8990
from graphdatascience.procedure_surface.arrow.node_embedding.fastrp_arrow_endpoints import FastRPArrowEndpoints
9091
from graphdatascience.procedure_surface.arrow.node_embedding.graphsage_predict_arrow_endpoints import (
9192
GraphSagePredictArrowEndpoints,
@@ -143,6 +144,10 @@ def set_show_progress(self, show_progress: bool) -> None:
143144
def graph(self) -> CatalogArrowEndpoints:
144145
return CatalogArrowEndpoints(self._arrow_client, self._db_client, show_progress=self._show_progress)
145146

147+
@property
148+
def config(self) -> ConfigArrowEndpoints:
149+
return ConfigArrowEndpoints(self._arrow_client)
150+
146151
## Algorithms
147152

148153
@property

0 commit comments

Comments
 (0)