impala-shell should handle broken pipe error gracefully

$ impala-shell.sh -d tpcds_parquet -q "show partitions store_sales;" | head -n 10 > /tmp/test.out
Starting Impala Shell with no authentication using Python 2.7.16
Warning: live_progress only applies to interactive shell sessions, and is being skipped for now.
Opened TCP connection to localhost:21050
Connected to localhost:21050
Server version: impalad version 4.4.0-SNAPSHOT DEBUG (build 67ed00560d0df2282731dc47a58fb3d679ffb99b)
Query: use `tpcds_parquet`
Query: use `tpcds_parquet`
Query: show partitions store_sales
Unknown Exception : [Errno 32] Broken pipe
Traceback (most recent call last):
  File "/home/rsuminto/workspace/impala/shell/impala_shell.py", line 1389, in _execute_stmt
    self.output_stream.write(rows)
  File "/home/rsuminto/workspace/impala/shell/shell_output.py", line 187, in write
    print(formatted_data)
IOError: [Errno 32] Broken pipe
Could not execute command: show partitions store_sales