diff --git a/libs/community/langchain_community/document_loaders/athena.py b/libs/community/langchain_community/document_loaders/athena.py index 1e33062c19..fc3c947dea 100644 --- a/libs/community/langchain_community/document_loaders/athena.py +++ b/libs/community/langchain_community/document_loaders/athena.py @@ -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)