From d34558f8b58fcc902084d79f7a23e0d3f6510f3c Mon Sep 17 00:00:00 2001
From: Adrian Macal <amacal@id5.io>
Date: Thu, 5 Sep 2024 16:26:37 +0200
Subject: [PATCH] Add work_group_enforced into AthenaCredentials

---
 dbt/adapters/athena/connections.py | 2 ++
 1 file changed, 2 insertions(+)

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",