From 9af14a5a1c036c859417b00ccbaf6db71fecdf62 Mon Sep 17 00:00:00 2001 From: Onkar Ravgan Date: Tue, 5 Sep 2023 18:18:16 +0530 Subject: [PATCH] fixed either in lineage req (#13088) --- .../ingestion/source/database/dbt/metadata.py | 28 ++++++++++--------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/ingestion/src/metadata/ingestion/source/database/dbt/metadata.py b/ingestion/src/metadata/ingestion/source/database/dbt/metadata.py index de365accef8..ef5ad1fd558 100644 --- a/ingestion/src/metadata/ingestion/source/database/dbt/metadata.py +++ b/ingestion/src/metadata/ingestion/source/database/dbt/metadata.py @@ -606,19 +606,21 @@ class DbtSource(DbtServiceSource): entity_list=from_es_result, fetch_multiple_entities=False ) if from_entity and to_entity: - yield AddLineageRequest( - edge=EntitiesEdge( - fromEntity=EntityReference( - id=from_entity.id.__root__, - type="table", - ), - toEntity=EntityReference( - id=to_entity.id.__root__, - type="table", - ), - lineageDetails=LineageDetails( - source=LineageSource.DbtLineage - ), + yield Either( + right=AddLineageRequest( + edge=EntitiesEdge( + fromEntity=EntityReference( + id=from_entity.id.__root__, + type="table", + ), + toEntity=EntityReference( + id=to_entity.id.__root__, + type="table", + ), + lineageDetails=LineageDetails( + source=LineageSource.DbtLineage + ), + ) ) )