fix(ingest/athena): Fixing db container id (#8689)

This commit is contained in:
Tamas Nemeth 2023-08-30 22:12:02 +02:00 committed by GitHub
parent 026f7abe9c
commit c193b1dc70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -192,15 +192,12 @@ class AthenaSource(SQLAlchemySource):
database: str,
extra_properties: Optional[Dict[str, Any]] = None,
) -> Iterable[MetadataWorkUnit]:
database_container_key = gen_database_key(
database,
platform=self.platform,
platform_instance=self.config.platform_instance,
env=self.config.env,
database_container_key = self.get_database_container_key(
db_name=database, schema=schema
)
yield from gen_database_container(
database=database,
database=database_container_key.database,
database_container_key=database_container_key,
sub_types=[DatasetContainerSubTypes.DATABASE],
domain_registry=self.domain_registry,