Read the ack message for each batch

neo4j-field · Oct 23, 2024 · 2340296 · 2340296
1 parent dfd3828
commit 2340296
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/neo4j_arrow/_client.py b/src/neo4j_arrow/_client.py
@@ -296,9 +296,9 @@ def _write_batches(
                 for batch in batches:
                     mapped_batch = fn(batch)
                     self._write_batch_with_retries(mapped_batch, writer)
+                    metadata_reader.read() # read the ack message
                     n_rows += batch.num_rows
                     n_bytes += batch.get_total_buffer_size()
-            metadata_reader.read()
         except Exception as e:
             raise error.interpret(e)
         return n_rows, n_bytes