From 757bc74bfde84c74f2c00aa9310a58e98f9630ed Mon Sep 17 00:00:00 2001
From: Roshan Punnoose <Roshan.Punnoose@jhuapl.edu>
Date: Thu, 3 Aug 2023 14:28:12 -0400
Subject: [PATCH] Adding ResultConfiguration for specific output location for
 Athena

---
 minerva/athena.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/minerva/athena.py b/minerva/athena.py
index e6d927d..ac2ccc5 100644
--- a/minerva/athena.py
+++ b/minerva/athena.py
@@ -44,11 +44,14 @@ class Execute:
         return self.sql
 
     def run(self):
+        config = {"OutputLocation": self.athena.output}
         if self.params:
             resp  = self.client.start_query_execution(QueryString=self.query(),
+                                                      ResultConfiguration=config,
                                                       ExecutionParameters=self.params)
         else:
-            resp  = self.client.start_query_execution(QueryString=self.query())
+            resp  = self.client.start_query_execution(QueryString=self.query(),
+                                                      ResultConfiguration=config)
 
         self.query_id = resp['QueryExecutionId']
         return resp