From 331f4db0ba1545e35f65bbadffb4a4e8f45c45d0 Mon Sep 17 00:00:00 2001 From: Tom Forbes Date: Sun, 20 Oct 2024 12:09:39 +0100 Subject: [PATCH] Use nbytes --- src/pypi_data/combine_parquet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/pypi_data/combine_parquet.py b/src/pypi_data/combine_parquet.py index 2e6948fc..7de5b0a6 100644 --- a/src/pypi_data/combine_parquet.py +++ b/src/pypi_data/combine_parquet.py @@ -36,7 +36,7 @@ def append_buffer( def buffer_mem_size(buffer: Deque[tuple[tuple[int, str], RecordBatch]]) -> int: - return sum(batch.get_total_buffer_size() for (_, _), batch in buffer) + return sum(batch.nbytes for (_, _), batch in buffer) def buffer_at_capacity(size: int) -> bool: