fix other bad maps

Signed-off-by: Jonathan Shi <[email protected]>

Author: noloerino

modin/core/dataframe/pandas/interchange/dataframe_protocol/column.py

@@ -292,7 +292,7 @@ def get_chunks(
         new_lengths = [chunksize] * n_chunks
         new_lengths[-1] = n_rows % n_chunks + new_lengths[-1]
 
-        new_partitions = self._col._partition_mgr_cls.map_partitions(
+        new_partitions = self._col._partition_mgr_cls.map_partitions_full_axis(
             self._col._partitions,
             lambda df: df,
             axis=0,

modin/core/dataframe/pandas/interchange/dataframe_protocol/dataframe.py

@@ -176,7 +176,7 @@ def get_chunks(
         new_lengths = [chunksize] * n_chunks
         new_lengths[-1] = n_rows % n_chunks + new_lengths[-1]
 
-        new_partitions = self._df._partition_mgr_cls.map_partitions(
+        new_partitions = self._df._partition_mgr_cls.map_partitions_full_axis(
             self._df._partitions,
             lambda df: df,
             axis=0,

modin/core/execution/ray/implementations/pandas_on_ray/io/io.py

@@ -112,8 +112,8 @@ def func(df):
 
         # Ensure that the metadata is syncrhonized
         qc._modin_frame._propagate_index_objs(axis=None)
-        result = qc._modin_frame.map(
-            func, axis=1, full_axis=True, new_index=[], new_columns=[]
+        result = qc._modin_frame.map_full_axis(
+            func, axis=1, new_index=[], new_columns=[]
         )
         # FIXME: we should be waiting for completion less expensively, maybe use _modin_frame.materialize()?
         result.to_pandas()  # blocking operation
@@ -230,7 +230,7 @@ def func(df, **kw):
         RayWrapper.materialize(signals.send.remote(0))
         # Ensure that the metadata is syncrhonized
         qc._modin_frame._propagate_index_objs(axis=None)
-        result = qc._modin_frame._partition_mgr_cls.map_partitions(
+        result = qc._modin_frame._partition_mgr_cls.map_partitions_full_axis(
             qc._modin_frame._partitions,
             func,
             axis=1,
@@ -309,7 +309,7 @@ def func(df, **kw):
 
         # Ensure that the metadata is synchronized
         qc._modin_frame._propagate_index_objs(axis=None)
-        result = qc._modin_frame._partition_mgr_cls.map_partitions(
+        result = qc._modin_frame._partition_mgr_cls.map_partitions_full_axis(
             qc._modin_frame._partitions,
             func,
             axis=1,