diff --git a/dataprofiler/labelers/data_processing.py b/dataprofiler/labelers/data_processing.py index 2213fd72..d53980a3 100644 --- a/dataprofiler/labelers/data_processing.py +++ b/dataprofiler/labelers/data_processing.py @@ -2047,9 +2047,9 @@ def process( elif aggregation_func == "random": num_labels = max(label_mapping.values()) + 1 random_state: random.Random = self._parameters["random_state"] - priority_order = np.array(list(range(num_labels))) - random_state.shuffle(priority_order) # type: ignore - self.priority_prediction(results, priority_order) + priority_order = list(range(num_labels)) + random_state.shuffle(priority_order) + self.priority_prediction(results, np.array(priority_order)) else: raise ValueError( f"`{aggregation_func}` is not a valid aggregation function"