From 405054ba200371f62413cb63563da28db50e9b51 Mon Sep 17 00:00:00 2001
From: Rui Xuan <ruixuantan@yahoo.com.sg>
Date: Thu, 20 Feb 2025 17:27:18 +0800
Subject: [PATCH] Add session token parameter to athena data source connection
 (#2095)

Co-authored-by: Milan Lukac <m1n0@users.noreply.github.com>
---
 soda/athena/soda/data_sources/athena_data_source.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/soda/athena/soda/data_sources/athena_data_source.py b/soda/athena/soda/data_sources/athena_data_source.py
index 359fc9e4c..d523dec6a 100644
--- a/soda/athena/soda/data_sources/athena_data_source.py
+++ b/soda/athena/soda/data_sources/athena_data_source.py
@@ -48,6 +48,7 @@ def connect(self):
                 catalog_name=self.catalog,
                 work_group=self.work_group,
                 schema_name=self.schema,
+                aws_session_token=self.aws_credentials.session_token,
             )
 
             return self.connection