expose catalog_name to the sql alchemy uri that is passed into pyathena

datahub-project · Aug 2, 2022 · 2ebd4f3 · 2ebd4f3
1 parent 823ae97
commit 2ebd4f3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/athena.py b/metadata-ingestion/src/datahub/ingestion/source/sql/athena.py
@@ -48,6 +48,9 @@ class AthenaConfig(SQLAlchemyConfig):
     work_group: str = pydantic.Field(
         description="The name of your Amazon Athena Workgroups"
     )
+    catalog_name: str = pydantic.Field(
+        default="awsdatacatalog", description="Athena Catalog Name"
+    )
 
     include_views = False  # not supported for Athena
 
@@ -61,6 +64,7 @@ def get_sql_alchemy_url(self):
             uri_opts={
                 "s3_staging_dir": self.s3_staging_dir,
                 "work_group": self.work_group,
+                "catalog_name": self.catalog_name,
             },
         )