PERF-#0000: avoid 'columnarize' function call

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>
modin-project · Nov 5, 2023 · 33684aa · 33684aa
1 parent 7a9415e
commit 33684aa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/modin/core/storage_formats/base/query_compiler.py b/modin/core/storage_formats/base/query_compiler.py
@@ -182,6 +182,7 @@ def default_to_pandas(self, pandas_op, *args, **kwargs):
 
     lazy_execution = False
     _shape_hint = None
+    _columnarize_if_needed = None
 
     # Metadata modification abstract methods
     def add_prefix(self, prefix, axis=1):

diff --git a/modin/pandas/series.py b/modin/pandas/series.py
@@ -137,7 +137,10 @@ def __init__(
                     )
                 )
             )._query_compiler
-        self._query_compiler = query_compiler.columnarize()
+        if query_compiler._columnarize_if_needed:
+            self._query_compiler = query_compiler.columnarize()
+        else:
+            self._query_compiler = query_compiler
         if name is not None:
             self.name = name