put correct dataset only (#13369)

* put correct dataset only

* update code

---------

Co-authored-by: hung.duong <hung.duong@be.com.vn>
Co-authored-by: Ayush Shah <ayush@getcollate.io>
Co-authored-by: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
This commit is contained in:
Hung Duong 2023-11-09 10:30:43 -08:00 committed by GitHub
parent 4ec585dca4
commit 575a4a0ebc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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