Added comment

modin-project · Sep 29, 2023 · 8842a0c · 8842a0c
1 parent 87a81bf
commit 8842a0c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/modin/core/dataframe/pandas/partitioning/partition_manager.py b/modin/core/dataframe/pandas/partitioning/partition_manager.py
@@ -257,6 +257,9 @@ def groupby_reduce(
             )
         else:
             mapped_partitions = cls.map_partitions(partitions, map_func)
+
+        # Assuming, that the output will not be larger than the input,
+        # keep the current number of partitions.
         num_splits = min(len(partitions), NPartitions.get())
         return cls.map_axis_partitions(
             axis,