community[patch]: remove print (#17435)

pull/15659/head
Bagatur 8 months ago committed by GitHub
parent 54fa78c887
commit f7e453971d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -82,14 +82,12 @@ class AthenaLoader(BaseLoader):
ResultConfiguration={"OutputLocation": self.s3_output_uri},
)
query_execution_id = response["QueryExecutionId"]
print(f"Query : {self.query}")
while True:
response = self.athena_client.get_query_execution(
QueryExecutionId=query_execution_id
)
state = response["QueryExecution"]["Status"]["State"]
if state == "SUCCEEDED":
print(f"State : {state}")
break
elif state == "FAILED":
resp_status = response["QueryExecution"]["Status"]
@ -98,8 +96,6 @@ class AthenaLoader(BaseLoader):
raise Exception(err)
elif state == "CANCELLED":
raise Exception("Query was cancelled by the user.")
else:
print(f"State : {state}")
time.sleep(1)
result_set = self._get_result_set(query_execution_id)

Loading…
Cancel
Save