diff --git a/cyclops/data/df/vectorized.py b/cyclops/data/df/vectorized.py index f759b37f3..380f273a7 100644 --- a/cyclops/data/df/vectorized.py +++ b/cyclops/data/df/vectorized.py @@ -143,9 +143,10 @@ def split_vectorized( seed=seed, ) - splits = [ - vec.split_by_indices(axes_list[i], index_splits) for i, vec in enumerate(vecs) # type: ignore - ] + splits = ( + vec.split_by_indices(axes_list[i], index_splits) # type: ignore + for i, vec in enumerate(vecs) + ) return tuple(splits)