fixing headers

CAVEconnectome · Aug 9, 2023 · d2d02fe · d2d02fe
1 parent 85c2d4f
commit d2d02fe
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/materializationengine/blueprints/client/utils.py b/materializationengine/blueprints/client/utils.py
@@ -88,7 +88,9 @@ def create_query_response(
             opt = pa.ipc.IpcWriteOptions(compression=compression)
             with pa.ipc.new_stream(sink, batch.schema, options=opt) as writer:
                 writer.write_batch(batch)
-            return send_file(BytesIO(sink.getvalue().to_pybytes()), "data.arrow")
+            response = send_file(BytesIO(sink.getvalue().to_pybytes()), "data.arrow")
+            response.headers = headers
+            return response
         # headers = add_warnings_to_headers(
         #     headers,
         #     [