Add work_group_enforced into AthenaCredentials

dbt-labs · Sep 5, 2024 · d34558f · d34558f
1 parent 9ddb520
commit d34558f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dbt/adapters/athena/connections.py b/dbt/adapters/athena/connections.py
@@ -56,6 +56,7 @@ class AthenaCredentials(Credentials):
     region_name: str
     endpoint_url: Optional[str] = None
     work_group: Optional[str] = None
+    work_group_enforced: Optional[bool] = None
     aws_profile_name: Optional[str] = None
     aws_access_key_id: Optional[str] = None
     aws_secret_access_key: Optional[str] = None
@@ -91,6 +92,7 @@ def _connection_keys(self) -> Tuple[str, ...]:
         return (
             "s3_staging_dir",
             "work_group",
+            "work_group_enforced",
             "region_name",
             "database",
             "schema",