add mlModel to DPIinput relationships entityTypes

datahub-project · Jan 20, 2025 · d4dc426 · d4dc426
1 parent b5066c6
commit d4dc426
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceInput.pdl b/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceInput.pdl
@@ -15,7 +15,7 @@ record DataProcessInstanceInput {
   @Relationship = {
     "/*": {
       "name": "Consumes",
-      "entityTypes": [ "dataset" ],
+      "entityTypes": [ "dataset", "mlModel" ],
       "isLineage": true
     }
   }