diff --git a/app_worker.py b/app_worker.py index efbd506..da9faa3 100644 --- a/app_worker.py +++ b/app_worker.py @@ -353,7 +353,7 @@ def get_num_statements_with_rows(query_multi, script): try: aws_multipart_upload(signed_s3_request, s3_key, stream_write_parquet(cols, rows)) - except (pa.ArrowNotImplementedError, pa.ArrowTypeError): + except pa.ArrowException: logger.exception('Unable to convert to parquet') # ... and as ODS with the results of each statement as a separate sheet