mssql: Implement mean operator

pydiverse · Jul 19, 2023 · 42bb190 · 42bb190
1 parent 506ff0c
commit 42bb190
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/pydiverse/transform/lazy/sql_table/dialects/mssql.py b/src/pydiverse/transform/lazy/sql_table/dialects/mssql.py
@@ -2,6 +2,7 @@
 
 import sqlalchemy as sa
 
+from pydiverse.transform.core import ops
 from pydiverse.transform.core.util import OrderingDescriptor
 from pydiverse.transform.lazy.sql_table.sql_table import SQLTableImpl
 
@@ -25,3 +26,10 @@ def _order_col(
 
         order_by_expressions.append(col.asc() if ordering.asc else col.desc())
         return order_by_expressions
+
+
+with MSSqlTableImpl.op(ops.Mean()) as op:
+
+    @op.auto
+    def _mean(x):
+        return sa.func.AVG(sa.cast(x, sa.Float()))