fix lint + default execution

modin-project · Sep 12, 2024 · 058433a · 058433a
1 parent 90fb128
commit 058433a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/modin/core/storage_formats/base/query_compiler.py b/modin/core/storage_formats/base/query_compiler.py
@@ -652,7 +652,7 @@ def series_eq(self, other, **kwargs):
             self,
             other=other,
             squeeze_self=True,
-            squeeze_other=True,
+            squeeze_other=isinstance(other, pandas.DataFrame),
             **kwargs,
         )
 
@@ -699,7 +699,7 @@ def series_ge(self, other, **kwargs):
             self,
             other=other,
             squeeze_self=True,
-            squeeze_other=True,
+            squeeze_other=isinstance(other, pandas.DataFrame),
             **kwargs,
         )
 
@@ -714,7 +714,7 @@ def series_gt(self, other, **kwargs):
             self,
             other=other,
             squeeze_self=True,
-            squeeze_other=True,
+            squeeze_other=isinstance(other, pandas.DataFrame),
             **kwargs,
         )
 
@@ -729,7 +729,7 @@ def series_le(self, other, **kwargs):
             self,
             other=other,
             squeeze_self=True,
-            squeeze_other=True,
+            squeeze_other=isinstance(other, pandas.DataFrame),
             **kwargs,
         )
 
@@ -744,7 +744,7 @@ def series_lt(self, other, **kwargs):
             self,
             other=other,
             squeeze_self=True,
-            squeeze_other=True,
+            squeeze_other=isinstance(other, pandas.DataFrame),
             **kwargs,
         )
 
@@ -868,7 +868,7 @@ def series_ne(self, other, **kwargs):
             self,
             other=other,
             squeeze_self=True,
-            squeeze_other=True,
+            squeeze_other=isinstance(other, pandas.DataFrame),
             **kwargs,
         )
 

diff --git a/modin/core/storage_formats/pandas/query_compiler.py b/modin/core/storage_formats/pandas/query_compiler.py
@@ -256,6 +256,15 @@ def caller(df, *args, **kwargs):
 def _series_logical_binop(func):
     """
     Build a callable function to pass to Binary.register for Series logical operators.
+
+    Parameters
+    ----------
+    func : callable
+        Binary operator method of pandas.Series to be applied.
+
+    Returns
+    -------
+    callable
     """
     return lambda x, y, **kwargs: func(
         x.squeeze(axis=1),