revert outputs

yoonhyejin · yoonhyejin · commit efad2f7de977 · 2025-03-14T17:41:23.000+09:00
diff --git a/metadata-ingestion/src/datahub/ingestion/source/mlflow.py b/metadata-ingestion/src/datahub/ingestion/source/mlflow.py
@@ -397,7 +397,12 @@ def _get_run_workunits(
             model_version_urn = self._make_ml_model_urn(model_versions[0])
             yield MetadataChangeProposalWrapper(
                 entityUrn=str(data_process_instance.urn),
-                aspect=DataProcessInstanceOutputClass(outputs=[model_version_urn]),
+                aspect=DataProcessInstanceOutputClass(
+                    outputs=[],
+                    outputEdges=[
+                        EdgeClass(destinationUrn=model_version_urn),
+                    ],
+                ),
             ).as_workunit()
 
         metrics = self._get_run_metrics(run)
diff --git a/metadata-ingestion/tests/integration/mlflow/mlflow_mcps_golden.json b/metadata-ingestion/tests/integration/mlflow/mlflow_mcps_golden.json
@@ -308,8 +308,11 @@
     "aspectName": "dataProcessInstanceOutput",
     "aspect": {
         "json": {
-            "outputs": [
-                "urn:li:mlModel:(urn:li:dataPlatform:mlflow,test-model_1,PROD)"
+            "outputs": [],
+            "outputEdges": [
+                {
+                    "destinationUrn": "urn:li:mlModel:(urn:li:dataPlatform:mlflow,test-model_1,PROD)"
+                }
             ]
         }
     },

Original file line number	Diff line number	Diff line change
`@@ -308,8 +308,11 @@`
`308`	`308`	`"aspectName": "dataProcessInstanceOutput",`
`309`	`309`	`"aspect": {`
`310`	`310`	`"json": {`
`311`		`- "outputs": [`
`312`		`- "urn:li:mlModel:(urn:li:dataPlatform:mlflow,test-model_1,PROD)"`
	`311`	`+ "outputs": [],`
	`312`	`+ "outputEdges": [`
	`313`	`+ {`
	`314`	`+ "destinationUrn": "urn:li:mlModel:(urn:li:dataPlatform:mlflow,test-model_1,PROD)"`
	`315`	`+ }`
`313`	`316`	`]`
`314`	`317`	`}`
`315`	`318`	`},`