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

@@ -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