PERF-#6373: preserve dtypes cache for '_repartition' (#6376)

Signed-off-by: Anatoly Myachev <[email protected]>
modin-project · Jul 13, 2023 · d6a6cf3 · d6a6cf3
1 parent b0e15d3
commit d6a6cf3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/modin/core/storage_formats/base/query_compiler.py b/modin/core/storage_formats/base/query_compiler.py
@@ -6593,6 +6593,7 @@ def repartition(self, axis=None):
                     lambda df: df,
                     new_index=self._modin_frame.copy_index_cache(),
                     new_columns=self._modin_frame.copy_columns_cache(),
+                    dtypes=self._modin_frame.copy_dtypes_cache(),
                     keep_partitioning=False,
                     sync_labels=False,
                 )