Skip to content

Commit c9d2a8c

Browse files
committed
fixed bug in memory calculation
1 parent 6d78e60 commit c9d2a8c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,7 @@ def split_dataframe(dataframe, chunk_size):
1515
current_size = 0
1616

1717
for index, row in dataframe.iterrows():
18-
row_size = row.memory_usage(deep=True).sum()
19-
18+
row_size = row.memory_usage(deep=True)
2019
if current_size + row_size > chunk_size:
2120
chunks.append(current_chunk)
2221
current_chunk = pd.DataFrame(columns=dataframe.columns)

0 commit comments

Comments
 (0)