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