Remove hax

Signed-off-by: Devin Petersohn <[email protected]>
modin-project · Jul 18, 2024 · 5800e3e · 5800e3e
1 parent 0019245
commit 5800e3e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/modin/polars/dataframe.py b/modin/polars/dataframe.py
@@ -704,7 +704,11 @@ def item(self, row: int | None = None, column: str | int | None = None) -> Any:
             column = 0
         if isinstance(column, str):
             column = self.columns.index(column)
-        return self.to_pandas().iloc[row, column]
+        return (
+            self._query_compiler.take_2d_labels(row, column)
+            .to_pandas()
+            .squeeze(axis=None)
+        )
 
     def iter_columns(self) -> Iterator["Series"]:
         """