Correction

Signed-off-by: Mohit Singla <[email protected]>

Author: mohitsingla-db

src/databricks/sql/thrift_backend.py

@@ -717,11 +717,9 @@ def _results_message_to_execute_response(self, resp, operation_state):
             .serialize()
             .to_pybytes()
         )
-
         are_arrow_results_compressed = (
             t_result_set_metadata_resp and t_result_set_metadata_resp.lz4Compressed
         )
-
         if direct_results and direct_results.resultSet:
             assert direct_results.resultSet.results.startRowOffset == 0
             assert direct_results.resultSetMetadata
@@ -940,6 +938,9 @@ def fetch_results(
                     expected_row_start_offset, resp.results.startRowOffset
                 )
             )
+        are_arrow_results_compressed = (
+            resp.resultSetMetadata and resp.resultSetMetadata.lz4Compressed
+        )
         arrow_results, n_rows = self._create_arrow_table(
             resp.results, are_arrow_results_compressed, arrow_schema_bytes, description
         )