Skip to content

Commit debec39

Browse files
Chore: Remove warning when creating chunks (#60)
1 parent 68aed00 commit debec39

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

data_validation_framework/util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def apply_to_df(df, func, *args, nb_processes=None, redirect_stdout=None, **kwar
107107
# Parallel computation
108108
is_parallel = True
109109
nb_chunks = min(nb_jobs, nb_processes)
110-
chunks = enumerate(np.array_split(df, nb_chunks))
110+
chunks = enumerate(df.loc[i] for i in np.array_split(df.index, nb_chunks))
111111
tqdm_queue = multiprocessing.Queue()
112112
message_queue = multiprocessing.Queue()
113113

0 commit comments

Comments
 (0)