From bfe77ed6d73b7f24d8b3e372df7994923ffbcf34 Mon Sep 17 00:00:00 2001 From: Dmitry Chigarev Date: Wed, 31 Jan 2024 16:28:28 +0100 Subject: [PATCH] FIX-#6897: Preprocess kernel function that alignes columns in groupby (#6898) Signed-off-by: Dmitry Chigarev --- modin/core/dataframe/pandas/dataframe/dataframe.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modin/core/dataframe/pandas/dataframe/dataframe.py b/modin/core/dataframe/pandas/dataframe/dataframe.py index 6ed18805336..a67db9a8cab 100644 --- a/modin/core/dataframe/pandas/dataframe/dataframe.py +++ b/modin/core/dataframe/pandas/dataframe/dataframe.py @@ -3897,7 +3897,9 @@ def compute_aligned_columns(*dfs, initial_columns=None): # aligned columns parts = result._partitions.flatten() aligned_columns = parts[0].apply( - compute_aligned_columns, + # TODO: unidist on MPI execution requires for this function to be preprocessed, + # otherwise, the execution fails. Look into the issue later. + self._partition_mgr_cls.preprocess_func(compute_aligned_columns), *[part._data for part in parts[1:]], initial_columns=self.columns, )