fix list RHS

modin-project · 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)