From d2deb38766712b92f70cca0b059d9c97e6b05922 Mon Sep 17 00:00:00 2001 From: Pere Miquel Brull Date: Thu, 15 Dec 2022 14:04:59 +0100 Subject: [PATCH] Fix edge nodes (#9274) --- .../src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py b/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py index 6169a6fc2af..9e2663d435b 100644 --- a/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py +++ b/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py @@ -79,10 +79,10 @@ class OMetaMlModelMixin(OMetaLineageMixin): AddLineageRequest( description="MlModel uses FeatureSource", edge=EntitiesEdge( - fromEntity=self.get_entity_reference( + fromEntity=entity_ref, + toEntity=self.get_entity_reference( entity=MlModel, fqn=model.fullyQualifiedName ), - toEntity=entity_ref, ), ) )