HEL/midas
1
0
Fork 0
Code Issues Pull Requests Releases Activity

feat(checker): add comparison binary ops on frames

Browse Source
This commit is contained in:
Louis Heredero
2026-07-03 00:57:27 +02:00
parent 6e06f9078e
commit d8c0b17512
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
midas/checker/frames/frame_methods.py
View File

@@ -203,6 +203,30 @@ class FrameMethodRegistry(MethodRegistry[Call]):
def pow(self, call: Call) -> Type:
return self._element_wise(call, "__pow__")
@method("lt", "__lt__")
def lt(self, call: Call) -> Type:
return self._element_wise(call, "__lt__")
@method("gt", "__gt__")
def gt(self, call: Call) -> Type:
return self._element_wise(call, "__gt__")
@method("le", "__le__")
def le(self, call: Call) -> Type:
return self._element_wise(call, "__le__")
@method("ge", "__ge__")
def ge(self, call: Call) -> Type:
return self._element_wise(call, "__ge__")
@method("ne", "__ne__")
def ne(self, call: Call) -> Type:
return self._element_wise(call, "__ne__")
@method("eq", "__eq__")
def eq(self, call: Call) -> Type:
return self._element_wise(call, "__eq__")
@method()
def mean(self, call: Call) -> Type:
with_axis = Function(