Use nbytes

pypi-data · Oct 20, 2024 · 331f4db · 331f4db
1 parent 8f64e40
commit 331f4db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: