Add mutateProperty

Co-authored-by: Florentin Dörre <florentin.dorre@neo4j.com>
Co-authored-by: Jacob Sznajdman <breakanalysis@gmail.com>
Co-authored-by: Olga Razvenskaia <olga.razvenskaia@neo4j.com>

Author: 4 people

graphdatascience/gnn/gnn_nc_runner.py

@@ -45,14 +45,16 @@ 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,
+            "mutateProperty": mutateProperty
         }
         if target_node_label:
             mlConfigMap["targetNodeLabel"] = target_node_label