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, database: str,
extra_properties: Optional[Dict[str, Any]] = None, extra_properties: Optional[Dict[str, Any]] = None,
) -> Iterable[MetadataWorkUnit]: ) -> Iterable[MetadataWorkUnit]:
database_container_key = gen_database_key( database_container_key = self.get_database_container_key(
database, db_name=database, schema=schema
platform=self.platform,
platform_instance=self.config.platform_instance,
env=self.config.env,
) )
yield from gen_database_container( yield from gen_database_container(
database=database, database=database_container_key.database,
database_container_key=database_container_key, database_container_key=database_container_key,
sub_types=[DatasetContainerSubTypes.DATABASE], sub_types=[DatasetContainerSubTypes.DATABASE],
domain_registry=self.domain_registry, domain_registry=self.domain_registry,