fix list RHS

noloerino · Sep 12, 2024 · 90fb128 · 90fb128
1 parent 763e09a
commit 90fb128
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/modin/core/storage_formats/pandas/query_compiler.py b/modin/core/storage_formats/pandas/query_compiler.py
@@ -258,7 +258,9 @@ def _series_logical_binop(func):
     Build a callable function to pass to Binary.register for Series logical operators.
     """
     return lambda x, y, **kwargs: func(
-        x.squeeze(axis=1), y.squeeze(axis=1), **kwargs
+        x.squeeze(axis=1),
+        y.squeeze(axis=1) if isinstance(y, pandas.DataFrame) else y,
+        **kwargs,
     ).to_frame()
 
 

diff --git a/modin/tests/pandas/test_series.py b/modin/tests/pandas/test_series.py
@@ -5130,3 +5130,12 @@ def test_binary_with_fill_value_issue_7381(op):
     result_md = getattr(series_md, op)(rhs_md, fill_value=2)
     result_pd = getattr(series_pd, op)(rhs_pd, fill_value=2)
     df_equals(result_md, result_pd)
+
+
+@pytest.mark.parametrize("op", ["eq", "ge", "gt", "le", "lt", "ne"])
+def test_logical_binary_with_list(op):
+    series_md, series_pd = create_test_series([0, 1, 2])
+    rhs = [2, 1, 0]
+    result_md = getattr(series_md, op)(rhs)
+    result_pd = getattr(series_pd, op)(rhs)
+    df_equals(result_md, result_pd)