Fix edge nodes (#9274)

This commit is contained in:
Pere Miquel Brull 2022-12-15 14:04:59 +01:00 committed by GitHub
parent 366a9fbc7c
commit d2deb38766
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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