diff --git a/protocol/dataframe_protocol.py b/protocol/dataframe_protocol.py index adde1a48..4d9e6c19 100644 --- a/protocol/dataframe_protocol.py +++ b/protocol/dataframe_protocol.py @@ -494,5 +494,8 @@ def get_chunks(self, n_chunks: Optional[int] = None) -> Iterable["DataFrame"]: producer. If given, ``n_chunks`` must be a multiple of ``self.num_chunks()``, meaning the producer must subdivide each chunk before yielding it. + + Note that the producer must ensure that all columns are chunked the + same way. """ pass