Skip to content

Commit

Permalink
[Data] Fix bug where _StatsActor errors with PandasBlock (#40481)
Browse files Browse the repository at this point in the history
StatsActor errors if your dataset contains pandas blocks.

Signed-off-by: Balaji Veeramani <balaji@anyscale.com>
  • Loading branch information
bveeramani authored Oct 19, 2023
1 parent d8f2527 commit 318fd57
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/ray/data/_internal/pandas_block.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ def num_rows(self) -> int:
return self._table.shape[0]

def size_bytes(self) -> int:
return self._table.memory_usage(index=True, deep=True).sum()
return int(self._table.memory_usage(index=True, deep=True).sum())

def _zip(self, acc: BlockAccessor) -> "pandas.DataFrame":
r = self.to_pandas().copy(deep=False)
Expand Down

0 comments on commit 318fd57

Please sign in to comment.