Skip to content

Commit e3b6ca5

Browse files
committed
Dont expose write_to_result_store param
1 parent 1ef1c53 commit e3b6ca5

13 files changed

+10
-38
lines changed

graphdatascience/procedure_surface/api/community/clique_counting_endpoints.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,6 @@ def write(
150150
sudo: Optional[bool] = False,
151151
username: Optional[str] = None,
152152
write_concurrency: Optional[int] = None,
153-
write_to_result_store: Optional[bool] = False,
154153
) -> CliqueCountingWriteResult:
155154
"""
156155
Executes the clique counting algorithm and writes the results back to the database.
@@ -177,8 +176,6 @@ def write(
177176
The username to attribute the procedure run to
178177
write_concurrency : Optional[int], default=None
179178
The number of concurrent threads for write operations
180-
write_to_result_store : Optional[bool], default=False
181-
Whether to write to the result store
182179
183180
Returns
184181
-------
@@ -220,7 +217,7 @@ def estimate(
220217
class CliqueCountingMutateResult(BaseResult):
221218
compute_millis: int
222219
configuration: dict[str, Any]
223-
global_count: List[Any]
220+
global_count: List[int]
224221
mutate_millis: int
225222
node_properties_written: int
226223
pre_processing_millis: int
@@ -229,14 +226,14 @@ class CliqueCountingMutateResult(BaseResult):
229226
class CliqueCountingStatsResult(BaseResult):
230227
compute_millis: int
231228
configuration: dict[str, Any]
232-
global_count: List[Any]
229+
global_count: List[int]
233230
pre_processing_millis: int
234231

235232

236233
class CliqueCountingWriteResult(BaseResult):
237234
compute_millis: int
238235
configuration: dict[str, Any]
239-
global_count: List[Any]
236+
global_count: List[int]
240237
node_properties_written: int
241238
pre_processing_millis: int
242239
write_millis: int

graphdatascience/procedure_surface/api/community/kmeans_endpoints.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def mutate(
6464
node_labels : Optional[List[str]], default=None
6565
The node labels used to select nodes for this algorithm run
6666
number_of_restarts : Optional[int], default=1
67-
The number of times the algorithm should be restarted
67+
The number of times the algorithm should be restarted with different initial centers
6868
random_seed : Optional[int], default=None
6969
Random seed for reproducible results
7070
relationship_types : Optional[List[str]], default=None
@@ -133,7 +133,7 @@ def stats(
133133
node_labels : Optional[List[str]], default=None
134134
The node labels used to select nodes for this algorithm run
135135
number_of_restarts : Optional[int], default=1
136-
The number of times the algorithm should be restarted
136+
The number of times the algorithm should be restarted with different initial centers
137137
random_seed : Optional[int], default=None
138138
Random seed for reproducible results
139139
relationship_types : Optional[List[str]], default=None
@@ -202,7 +202,7 @@ def stream(
202202
node_labels : Optional[List[str]], default=None
203203
The node labels used to select nodes for this algorithm run
204204
number_of_restarts : Optional[int], default=1
205-
The number of times the algorithm should be restarted
205+
The number of times the algorithm should be restarted with different initial centers
206206
random_seed : Optional[int], default=None
207207
Random seed for reproducible results
208208
relationship_types : Optional[List[str]], default=None
@@ -244,7 +244,6 @@ def write(
244244
sudo: Optional[bool] = False,
245245
username: Optional[str] = None,
246246
write_concurrency: Optional[int] = None,
247-
write_to_result_store: Optional[bool] = False,
248247
) -> KMeansWriteResult:
249248
"""
250249
Executes the K-Means algorithm and writes the results back to the database.
@@ -276,7 +275,7 @@ def write(
276275
node_labels : Optional[List[str]], default=None
277276
The node labels used to select nodes for this algorithm run
278277
number_of_restarts : Optional[int], default=1
279-
The number of times the algorithm should be restarted
278+
The number of times the algorithm should be restarted with different initial centers
280279
random_seed : Optional[int], default=None
281280
Random seed for reproducible results
282281
relationship_types : Optional[List[str]], default=None
@@ -289,8 +288,6 @@ def write(
289288
The username to attribute the procedure run to
290289
write_concurrency : Optional[int], default=None
291290
The number of concurrent threads for write operations
292-
write_to_result_store : Optional[bool], default=False
293-
Whether to write to the result store
294291
295292
Returns
296293
-------
@@ -341,7 +338,7 @@ def estimate(
341338
node_labels : Optional[List[str]], default=None
342339
The node labels used to select nodes for this algorithm run
343340
number_of_restarts : Optional[int], default=1
344-
The number of times the algorithm should be restarted
341+
The number of times the algorithm should be restarted with different initial centers
345342
random_seed : Optional[int], default=None
346343
Random seed for reproducible results
347344
relationship_types : Optional[List[str]], default=None

graphdatascience/procedure_surface/api/community/labelpropagation_endpoints.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,6 @@ def write(
208208
sudo: Optional[bool] = False,
209209
username: Optional[str] = None,
210210
write_concurrency: Optional[int] = None,
211-
write_to_result_store: Optional[bool] = False,
212211
) -> LabelPropagationWriteResult:
213212
"""
214213
Executes the Label Propagation algorithm and writes the results back to the database.
@@ -247,8 +246,6 @@ def write(
247246
The username to attribute the procedure run to
248247
write_concurrency : Optional[int], default=None
249248
The number of concurrent threads for write operations
250-
write_to_result_store : Optional[bool], default=False
251-
Whether to write to the result store
252249
253250
Returns
254251
-------

graphdatascience/procedure_surface/api/community/leiden_endpoints.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,6 @@ def write(
245245
tolerance: Optional[float] = 1e-4,
246246
username: Optional[str] = None,
247247
write_concurrency: Optional[int] = None,
248-
write_to_result_store: Optional[bool] = False,
249248
) -> "LeidenWriteResult":
250249
"""
251250
Executes the Leiden community detection algorithm and writes the results back to the database.
@@ -292,9 +291,7 @@ def write(
292291
The username to attribute the procedure run to
293292
write_concurrency : Optional[int], default=None
294293
The number of concurrent threads for writing
295-
write_to_result_store : Optional[bool], default=False
296-
Whether to write to result store
297-
294+
298295
Returns
299296
-------
300297
LeidenWriteResult

graphdatascience/procedure_surface/arrow/clique_counting_arrow_endpoints.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,6 @@ def write(
120120
sudo: Optional[bool] = False,
121121
username: Optional[str] = None,
122122
write_concurrency: Optional[Any] = None,
123-
write_to_result_store: Optional[bool] = False,
124123
) -> CliqueCountingWriteResult:
125124
config = self._node_property_endpoints.create_base_config(
126125
G,
@@ -131,7 +130,6 @@ def write(
131130
relationship_types=relationship_types,
132131
sudo=sudo,
133132
username=username,
134-
write_to_result_store=write_to_result_store,
135133
)
136134

137135
result = self._node_property_endpoints.run_job_and_write(

graphdatascience/procedure_surface/arrow/kmeans_arrow_endpoints.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,6 @@ def write(
183183
sudo: Optional[bool] = False,
184184
username: Optional[str] = None,
185185
write_concurrency: Optional[Any] = None,
186-
write_to_result_store: Optional[bool] = False,
187186
) -> KMeansWriteResult:
188187
config = self._node_property_endpoints.create_base_config(
189188
G,
@@ -203,7 +202,6 @@ def write(
203202
seed_centroids=seed_centroids,
204203
sudo=sudo,
205204
username=username,
206-
write_to_result_store=write_to_result_store,
207205
)
208206

209207
result = self._node_property_endpoints.run_job_and_write(

graphdatascience/procedure_surface/arrow/labelpropagation_arrow_endpoints.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,6 @@ def write(
162162
sudo: Optional[bool] = False,
163163
username: Optional[str] = None,
164164
write_concurrency: Optional[int] = None,
165-
write_to_result_store: Optional[bool] = False,
166165
) -> LabelPropagationWriteResult:
167166
config = self._node_property_endpoints.create_base_config(
168167
G,
@@ -179,7 +178,6 @@ def write(
179178
seed_property=seed_property,
180179
sudo=sudo,
181180
username=username,
182-
write_to_result_store=write_to_result_store,
183181
)
184182

185183
result = self._node_property_endpoints.run_job_and_write(

graphdatascience/procedure_surface/arrow/leiden_arrow_endpoints.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,6 @@ def write(
187187
tolerance: Optional[float] = 1e-4,
188188
username: Optional[str] = None,
189189
write_concurrency: Optional[Any] = None,
190-
write_to_result_store: Optional[bool] = False,
191190
) -> LeidenWriteResult:
192191
config = self._node_property_endpoints.create_base_config(
193192
G,
@@ -208,7 +207,6 @@ def write(
208207
theta=theta,
209208
tolerance=tolerance,
210209
username=username,
211-
write_to_result_store=write_to_result_store,
212210
)
213211

214212
result = self._node_property_endpoints.run_job_and_write(

graphdatascience/procedure_surface/cypher/clique_counting_cypher_endpoints.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,6 @@ def write(
128128
sudo: Optional[bool] = False,
129129
username: Optional[str] = None,
130130
write_concurrency: Optional[int] = None,
131-
write_to_result_store: Optional[bool] = False,
132131
) -> CliqueCountingWriteResult:
133132
config = ConfigConverter.convert_to_gds_config(
134133
write_property=write_property,
@@ -140,7 +139,6 @@ def write(
140139
sudo=sudo,
141140
username=username,
142141
write_concurrency=write_concurrency,
143-
write_to_result_store=write_to_result_store,
144142
)
145143

146144
params = CallParameters(graph_name=G.name(), config=config)

graphdatascience/procedure_surface/cypher/kmeans_cypher_endpoints.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,6 @@ def write(
193193
sudo: Optional[bool] = False,
194194
username: Optional[str] = None,
195195
write_concurrency: Optional[int] = None,
196-
write_to_result_store: Optional[bool] = False,
197196
) -> KMeansWriteResult:
198197
config = ConfigConverter.convert_to_gds_config(
199198
node_property=node_property,
@@ -214,7 +213,6 @@ def write(
214213
sudo=sudo,
215214
username=username,
216215
write_concurrency=write_concurrency,
217-
write_to_result_store=write_to_result_store,
218216
)
219217

220218
params = CallParameters(graph_name=G.name(), config=config)

0 commit comments

Comments
 (0)