From 5b0c5c01ad1c22e086f0ced875bb8f1d98f13591 Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Fri, 26 Jun 2026 11:21:38 +0200
Subject: [PATCH] feat(checker): add mean method on frames

---
 midas/checker/frame_methods.py | 45 ++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/midas/checker/frame_methods.py b/midas/checker/frame_methods.py
index 96c868c..0220ba0 100644
--- a/midas/checker/frame_methods.py
+++ b/midas/checker/frame_methods.py
@@ -10,6 +10,8 @@ from midas.checker.types import (
     ColumnType,
     DataFrameType,
     Function,
+    OverloadedFunction,
+    TopType,
     Type,
     UnknownType,
     unfold_type,
@@ -85,6 +87,9 @@ class MethodRegistry(metaclass=_MethodRegistryMeta):
         self,
         call: Call,
     ) -> Type:
+        # TODO: support add with scalar, sequence, Series, dict
+        # TODO: check operation exists on inner column types
+
         new_columns: list[DataFrameType.Column] = []
 
         by_name: dict[str, DataFrameType.Column] = {}
@@ -151,3 +156,43 @@ class MethodRegistry(metaclass=_MethodRegistryMeta):
             )
             or UnknownType()
         )
+
+    @frame_method()
+    def mean(self, call: Call) -> Type:
+        with_axis = Function(
+            kw_args=[
+                Function.Argument(
+                    pos=0,
+                    name="axis",
+                    type=self.types.get_type("int"),
+                    required=False,
+                )
+            ],
+            returns=ColumnType(type=TopType()),
+        )
+        without_axis = Function(
+            kw_args=[
+                Function.Argument(
+                    pos=0,
+                    name="axis",
+                    type=self.types.get_type("None"),
+                    required=True,
+                )
+            ],
+            returns=TopType(),
+        )
+        overload = OverloadedFunction(
+            overloads=[
+                with_axis,
+                without_axis,
+            ]
+        )
+        return (
+            self.typer._get_call_result(
+                location=call.location,
+                callee=overload,
+                positional=call.positional,
+                keywords=call.keywords,
+            )
+            or UnknownType()
+        )