move to geo_ops

Author: tswast

bigframes/core/compile/sqlglot/expressions/geo_ops.py

@@ -82,3 +82,22 @@ def _(expr: TypedExpr) -> sge.Expression:
 @register_unary_op(ops.geo_y_op)
 def _(expr: TypedExpr) -> sge.Expression:
     return sge.func("SAFE.ST_Y", expr.expr)
+
+
+@register_unary_op(ops.StRegionStatsOp, pass_op=True)
+def _(
+    geography: TypedExpr,
+    op: ops.StRegionStatsOp,
+):
+    args = [geography.expr]  # TODO: get raster, band, include from op.
+    if op.options:
+        args.append(
+            sge.Anonymous(
+                this="_",
+                expressions=[
+                    sge.Identifier(this="OPTIONS"),
+                    sge.Anonymous(this="JSON", expressions=[sge.convert(op.options)]),
+                ],
+            )
+        )
+    return sge.func("ST_REGIONSTATS", *args)

bigframes/core/compile/sqlglot/scalar_compiler.py

@@ -22,7 +22,6 @@
 import bigframes.core.compile.sqlglot.sqlglot_ir as ir
 import bigframes.core.expression as ex
 import bigframes.operations as ops
-import bigframes.operations.geo_ops as geo_ops
 
 
 class ScalarOpCompiler:
@@ -228,22 +227,3 @@ def _add_parentheses(cls, expr: TypedExpr) -> TypedExpr:
 
 # Singleton compiler
 scalar_op_compiler = ScalarOpCompiler()
-
-
-@scalar_op_compiler.register_unary_op(geo_ops.StRegionStatsOp, pass_op=True)
-def compile_st_regionstats(
-    geography: TypedExpr,
-    op: geo_ops.StRegionStatsOp,
-):
-    args = [geography.expr]  # TODO: get raster, band, include from op.
-    if op.options:
-        args.append(
-            sge.Anonymous(
-                this="_",
-                expressions=[
-                    sge.Identifier(this="OPTIONS"),
-                    sge.Anonymous(this="JSON", expressions=[sge.convert(op.options)]),
-                ],
-            )
-        )
-    return sge.func("ST_REGIONSTATS", *args)