Skip to content

Commit 362735d

Browse files
committed
Remove mutateProperty from CRD
1 parent 877e796 commit 362735d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

graphdatascience/gnn/gnn_nc_runner.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def train(
3535
self._query_runner.run_query(
3636
"CALL gds.upload.graph($config)",
3737
params={
38-
"config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name, "mutateProperty": "TODO,REMOVE"},
38+
"config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name},
3939
},
4040
)
4141

@@ -54,7 +54,6 @@ def predict(
5454
"job_type": "predict",
5555
"nodeProperties": feature_properties,
5656
"relationshipTypes": relationship_types,
57-
##TODO cleanup the same from outside config
5857
"mutateProperty": mutateProperty
5958
}
6059
if target_node_label:
@@ -66,6 +65,6 @@ def predict(
6665
self._query_runner.run_query(
6766
"CALL gds.upload.graph($config)",
6867
params={
69-
"config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name, "mutateProperty": mutateProperty},
68+
"config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name},
7069
},
7170
) # type: ignore

0 commit comments

Comments
 (0)