From d8c0b17512fbf3de0a4d5dbf870ac1dde7d3d095 Mon Sep 17 00:00:00 2001 From: LordBaryhobal Date: Fri, 3 Jul 2026 00:57:27 +0200 Subject: [PATCH] feat(checker): add comparison binary ops on frames --- midas/checker/frames/frame_methods.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/midas/checker/frames/frame_methods.py b/midas/checker/frames/frame_methods.py index 4c413b4..c63e5b8 100644 --- a/midas/checker/frames/frame_methods.py +++ b/midas/checker/frames/frame_methods.py @@ -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(