Specify the write_property in all arrow algo endpoints

DarthMax · Mats-SX · commit 1464300e245c · 2025-09-18T15:48:10.000+02:00
diff --git a/graphdatascience/procedure_surface/arrow/articlerank_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/articlerank_arrow_endpoints.py
@@ -172,7 +172,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.articleRank", G, config, write_concurrency, concurrency
+            "v2/centrality.articleRank", G, config, write_concurrency, concurrency, write_property
         )
 
         return ArticleRankWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/articulationpoints_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/articulationpoints_arrow_endpoints.py
@@ -123,7 +123,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.articulationPoints", G, config, write_concurrency, concurrency
+            "v2/centrality.articulationPoints", G, config, write_concurrency, concurrency, write_property
         )
 
         return ArticulationPointsWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/betweenness_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/betweenness_arrow_endpoints.py
@@ -154,7 +154,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.betweenness", G, config, write_concurrency, concurrency
+            "v2/centrality.betweenness", G, config, write_concurrency, concurrency, write_property
         )
 
         return BetweennessWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/celf_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/celf_arrow_endpoints.py
@@ -147,7 +147,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.celf", G, config, write_concurrency=write_concurrency, concurrency=concurrency
+            "v2/centrality.celf", G, config, write_concurrency=write_concurrency, concurrency=concurrency, property_overwrites=write_property
         )
 
         return CelfWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/closeness_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/closeness_arrow_endpoints.py
@@ -134,7 +134,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.closeness", G, config, write_concurrency, concurrency
+            "v2/centrality.closeness", G, config, write_concurrency, concurrency, write_property
         )
 
         return ClosenessWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/closeness_harmonic_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/closeness_harmonic_arrow_endpoints.py
@@ -124,7 +124,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.harmonic", G, config, write_concurrency, concurrency
+            "v2/centrality.harmonic", G, config, write_concurrency, concurrency, write_property
         )
 
         return ClosenessHarmonicWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/degree_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/degree_arrow_endpoints.py
@@ -131,7 +131,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.degree", G, config, write_concurrency, concurrency
+            "v2/centrality.degree", G, config, write_concurrency, concurrency, write_property
         )
 
         return DegreeWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/eigenvector_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/eigenvector_arrow_endpoints.py
@@ -168,7 +168,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.eigenvector", G, config, write_concurrency, concurrency
+            "v2/centrality.eigenvector", G, config, write_concurrency, concurrency, write_property
         )
 
         return EigenvectorWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/graphsage_predict_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/graphsage_predict_arrow_endpoints.py
@@ -84,6 +84,7 @@ def write(
             config,
             write_concurrency,
             concurrency,
+            write_property
         )
 
         return GraphSageWriteResult(**raw_result)
diff --git a/graphdatascience/procedure_surface/arrow/hashgnn_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/hashgnn_arrow_endpoints.py
@@ -169,7 +169,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/embeddings.hashgnn", G, config, write_concurrency, concurrency
+            "v2/embeddings.hashgnn", G, config, write_concurrency, concurrency, write_property
         )
 
         return HashGNNWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/k1coloring_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/k1coloring_arrow_endpoints.py
@@ -144,7 +144,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/community.k1coloring", G, config, write_concurrency, concurrency
+            "v2/community.k1coloring", G, config, write_concurrency, concurrency, write_property
         )
 
         return K1ColoringWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/kcore_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/kcore_arrow_endpoints.py
@@ -119,7 +119,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/community.kcore", G, config, write_concurrency, concurrency
+            "v2/community.kcore", G, config, write_concurrency, concurrency, write_property
         )
         return KCoreWriteResult(**result)
 
diff --git a/graphdatascience/procedure_surface/arrow/louvain_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/louvain_arrow_endpoints.py
@@ -176,7 +176,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/community.louvain", G, config, write_concurrency, concurrency
+            "v2/community.louvain", G, config, write_concurrency, concurrency, write_property
         )
 
         return LouvainWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/node2vec_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/node2vec_arrow_endpoints.py
@@ -191,7 +191,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/embeddings.node2vec", G, config, write_concurrency, concurrency
+            "v2/embeddings.node2vec", G, config, write_concurrency, concurrency, write_property
         )
 
         return Node2VecWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/pagerank_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/pagerank_arrow_endpoints.py
@@ -163,7 +163,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/centrality.pageRank", G, config, write_concurrency, concurrency
+            "v2/centrality.pageRank", G, config, write_concurrency, concurrency, write_property
         )
         return PageRankWriteResult(**result)
 
diff --git a/graphdatascience/procedure_surface/arrow/scc_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/scc_arrow_endpoints.py
@@ -123,7 +123,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/community.scc", G, config, write_concurrency, concurrency
+            "v2/community.scc", G, config, write_concurrency, concurrency, write_property
         )
 
         return SccWriteResult(**result)
diff --git a/graphdatascience/procedure_surface/arrow/wcc_arrow_endpoints.py b/graphdatascience/procedure_surface/arrow/wcc_arrow_endpoints.py
@@ -151,7 +151,7 @@ def write(
         )
 
         result = self._node_property_endpoints.run_job_and_write(
-            "v2/community.wcc", G, config, write_concurrency, concurrency
+            "v2/community.wcc", G, config, write_concurrency, concurrency, write_property
         )
 
         return WccWriteResult(**result)

Original file line number	Diff line number	Diff line change
`@@ -172,7 +172,7 @@ def write(`
`172`	`172`	`)`
`173`	`173`
`174`	`174`	`result = self._node_property_endpoints.run_job_and_write(`
`175`		`- "v2/centrality.articleRank", G, config, write_concurrency, concurrency`
	`175`	`+ "v2/centrality.articleRank", G, config, write_concurrency, concurrency, write_property`
`176`	`176`	`)`
`177`	`177`
`178`	`178`	`return ArticleRankWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ def write(`
`123`	`123`	`)`
`124`	`124`
`125`	`125`	`result = self._node_property_endpoints.run_job_and_write(`
`126`		`- "v2/centrality.articulationPoints", G, config, write_concurrency, concurrency`
	`126`	`+ "v2/centrality.articulationPoints", G, config, write_concurrency, concurrency, write_property`
`127`	`127`	`)`
`128`	`128`
`129`	`129`	`return ArticulationPointsWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -154,7 +154,7 @@ def write(`
`154`	`154`	`)`
`155`	`155`
`156`	`156`	`result = self._node_property_endpoints.run_job_and_write(`
`157`		`- "v2/centrality.betweenness", G, config, write_concurrency, concurrency`
	`157`	`+ "v2/centrality.betweenness", G, config, write_concurrency, concurrency, write_property`
`158`	`158`	`)`
`159`	`159`
`160`	`160`	`return BetweennessWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -147,7 +147,7 @@ def write(`
`147`	`147`	`)`
`148`	`148`
`149`	`149`	`result = self._node_property_endpoints.run_job_and_write(`
`150`		`- "v2/centrality.celf", G, config, write_concurrency=write_concurrency, concurrency=concurrency`
	`150`	`+ "v2/centrality.celf", G, config, write_concurrency=write_concurrency, concurrency=concurrency, property_overwrites=write_property`
`151`	`151`	`)`
`152`	`152`
`153`	`153`	`return CelfWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -134,7 +134,7 @@ def write(`
`134`	`134`	`)`
`135`	`135`
`136`	`136`	`result = self._node_property_endpoints.run_job_and_write(`
`137`		`- "v2/centrality.closeness", G, config, write_concurrency, concurrency`
	`137`	`+ "v2/centrality.closeness", G, config, write_concurrency, concurrency, write_property`
`138`	`138`	`)`
`139`	`139`
`140`	`140`	`return ClosenessWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -124,7 +124,7 @@ def write(`
`124`	`124`	`)`
`125`	`125`
`126`	`126`	`result = self._node_property_endpoints.run_job_and_write(`
`127`		`- "v2/centrality.harmonic", G, config, write_concurrency, concurrency`
	`127`	`+ "v2/centrality.harmonic", G, config, write_concurrency, concurrency, write_property`
`128`	`128`	`)`
`129`	`129`
`130`	`130`	`return ClosenessHarmonicWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ def write(`
`131`	`131`	`)`
`132`	`132`
`133`	`133`	`result = self._node_property_endpoints.run_job_and_write(`
`134`		`- "v2/centrality.degree", G, config, write_concurrency, concurrency`
	`134`	`+ "v2/centrality.degree", G, config, write_concurrency, concurrency, write_property`
`135`	`135`	`)`
`136`	`136`
`137`	`137`	`return DegreeWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -168,7 +168,7 @@ def write(`
`168`	`168`	`)`
`169`	`169`
`170`	`170`	`result = self._node_property_endpoints.run_job_and_write(`
`171`		`- "v2/centrality.eigenvector", G, config, write_concurrency, concurrency`
	`171`	`+ "v2/centrality.eigenvector", G, config, write_concurrency, concurrency, write_property`
`172`	`172`	`)`
`173`	`173`
`174`	`174`	`return EigenvectorWriteResult(**result)`
Original file line number	Diff line number	Diff line change
`@@ -84,6 +84,7 @@ def write(`
`84`	`84`	`config,`
`85`	`85`	`write_concurrency,`
`86`	`86`	`concurrency,`
	`87`	`+ write_property`
`87`	`88`	`)`
`88`	`89`
`89`	`90`	`return GraphSageWriteResult(**raw_result)`
Original file line number	Diff line number	Diff line change
`@@ -169,7 +169,7 @@ def write(`
`169`	`169`	`)`
`170`	`170`
`171`	`171`	`result = self._node_property_endpoints.run_job_and_write(`
`172`		`- "v2/embeddings.hashgnn", G, config, write_concurrency, concurrency`
	`172`	`+ "v2/embeddings.hashgnn", G, config, write_concurrency, concurrency, write_property`
`173`	`173`	`)`
`174`	`174`
`175`	`175`	`return HashGNNWriteResult(**result)`