diff --git a/daft/io/_delta_lake.py b/daft/io/_delta_lake.py index 6bf8e764f4..b29c5cfc5c 100644 --- a/daft/io/_delta_lake.py +++ b/daft/io/_delta_lake.py @@ -72,7 +72,13 @@ def read_delta_lake( aws_secret_access_key=s3_config.access_key, aws_session_token=s3_config.session_token, ) - glue_table = glue.get_table(DatabaseName=table.database_name, Name=table.table_name) + if table.catalog_id is not None: + # All cross account access, table.catalog_id should be the target account id + glue_table = glue.get_table(CatalogId=table.catalog_id, DatabaseName=table.database_name, + Name=table.table_name) + else: + glue_table = glue.get_table(DatabaseName=table.database_name, Name=table.table_name) + # TODO(Clark): Fetch more than just the table URI from Glue Data Catalog. table_uri = glue_table["Table"]["StorageDescriptor"]["Location"] elif table.catalog == DataCatalogType.UNITY: