From 575a4a0ebc7f0b0a362fe9cd100423c8d4478470 Mon Sep 17 00:00:00 2001 From: Hung Duong <77838344+duongphannamhung@users.noreply.github.com> Date: Thu, 9 Nov 2023 10:30:43 -0800 Subject: [PATCH] put correct dataset only (#13369) * put correct dataset only * update code --------- Co-authored-by: hung.duong Co-authored-by: Ayush Shah Co-authored-by: Mayur Singal <39544459+ulixius9@users.noreply.github.com> --- .../src/metadata/ingestion/lineage/sql_lineage.py | 15 ++------------- 1 file changed, 2 insertions(+), 13 deletions(-) diff --git a/ingestion/src/metadata/ingestion/lineage/sql_lineage.py b/ingestion/src/metadata/ingestion/lineage/sql_lineage.py index 7851a1320eb..4cae5a53948 100644 --- a/ingestion/src/metadata/ingestion/lineage/sql_lineage.py +++ b/ingestion/src/metadata/ingestion/lineage/sql_lineage.py @@ -176,19 +176,8 @@ def get_table_entities_from_query( table_entities = search_table_entities( metadata=metadata, service_name=service_name, - database=database_query, - database_schema=schema_query, - table=table, - ) - - if table_entities: - return table_entities - - table_entities = search_table_entities( - metadata=metadata, - service_name=service_name, - database=database_name, - database_schema=database_schema, + database=database_query if database_query else database_name, + database_schema=schema_query if schema_query else database_schema, table=table, )