diff --git a/dbt/adapters/athena/connections.py b/dbt/adapters/athena/connections.py index 8536ae3c..5232abe2 100644 --- 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",