Skip to content

Commit

Permalink
Address comment
Browse files Browse the repository at this point in the history
Signed-off-by: Devin Petersohn <devin.petersohn@snowflake.com>
  • Loading branch information
sfc-gh-dpetersohn committed Jul 24, 2024
1 parent f89c66a commit 5a072ea
Showing 1 changed file with 16 additions and 16 deletions.
32 changes: 16 additions & 16 deletions modin/polars/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,112 +44,112 @@ def __constructor__(self):
def __eq__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 45 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L45

Added line #L45 was not covered by tests
_query_compiler=self._query_compiler.eq(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __ne__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 52 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L52

Added line #L52 was not covered by tests
_query_compiler=self._query_compiler.ne(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __add__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 59 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L59

Added line #L59 was not covered by tests
_query_compiler=self._query_compiler.add(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __sub__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 66 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L66

Added line #L66 was not covered by tests
_query_compiler=self._query_compiler.sub(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __mul__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 73 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L73

Added line #L73 was not covered by tests
_query_compiler=self._query_compiler.mul(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __truediv__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 80 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L80

Added line #L80 was not covered by tests
_query_compiler=self._query_compiler.truediv(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __floordiv__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 87 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L87

Added line #L87 was not covered by tests
_query_compiler=self._query_compiler.floordiv(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __mod__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 94 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L94

Added line #L94 was not covered by tests
_query_compiler=self._query_compiler.mod(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __pow__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 101 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L101

Added line #L101 was not covered by tests
_query_compiler=self._query_compiler.pow(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __and__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 108 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L108

Added line #L108 was not covered by tests
_query_compiler=self._query_compiler.__and__(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __or__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 115 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L115

Added line #L115 was not covered by tests
_query_compiler=self._query_compiler.__or__(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __xor__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 122 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L122

Added line #L122 was not covered by tests
_query_compiler=self._query_compiler.__xor__(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __lt__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 129 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L129

Added line #L129 was not covered by tests
_query_compiler=self._query_compiler.lt(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __le__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 136 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L136

Added line #L136 was not covered by tests
_query_compiler=self._query_compiler.le(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __gt__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 143 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L143

Added line #L143 was not covered by tests
_query_compiler=self._query_compiler.gt(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

def __ge__(self, other) -> "BasePolarsDataset":
return self.__constructor__(

Check warning on line 150 in modin/polars/base.py

View check run for this annotation

Codecov / codecov/patch

modin/polars/base.py#L150

Added line #L150 was not covered by tests
_query_compiler=self._query_compiler.ge(
other._query_compiler if hasattr(other, "_query_compiler") else other
other._query_compiler if isinstance(other, BasePolarsDataset) else other
)
)

Expand Down

0 comments on commit 5a072ea

Please sign in to comment.