WIP add mutateProperty

Co-authored-by: Florentin Dörre <[email protected]>
Co-authored-by: Jacob Sznajdman <[email protected]>
Co-authored-by: Olga Razvenskaia <[email protected]>

Author: 4 people

graphdatascience/gnn/gnn_nc_runner.py

@@ -35,7 +35,7 @@ def train(
         self._query_runner.run_query(
             "CALL gds.upload.graph($config)",
             params={
-                "config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name},
+                "config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name, "mutateProperty": "TODO,REMOVE"},
             },
         )
 
@@ -45,14 +45,17 @@ def predict(
         model_name: str,
         feature_properties: List[str],
         relationship_types: List[str],
+        mutateProperty: str,
         target_node_label: str = None,
         node_labels: List[str] = None,
     ) -> "Series[Any]":  # noqa: F821
         mlConfigMap = {
             "featureProperties": feature_properties,
             "job_type": "predict",
             "nodeProperties": feature_properties,
-            "relationshipTypes": relationship_types
+            "relationshipTypes": relationship_types,
+            ##TODO cleanup the same from outside config
+            "mutateProperty": mutateProperty
         }
         if target_node_label:
             mlConfigMap["targetNodeLabel"] = target_node_label
@@ -63,6 +66,6 @@ def predict(
         self._query_runner.run_query(
             "CALL gds.upload.graph($config)",
             params={
-                "config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name},
+                "config": {"mlTrainingConfig": mlTrainingConfig, "graphName": graph_name, "modelName": model_name, "mutateProperty": mutateProperty},
             },
         )  # type: ignore