From f21c5232381e3fc33714806b554e56f8768dbcd2 Mon Sep 17 00:00:00 2001 From: jialuo Date: Fri, 14 Nov 2025 00:02:36 +0000 Subject: [PATCH] Refactor: Use sge.func for ML.DISTANCE --- .../compile/sqlglot/expressions/numeric_ops.py | 18 +++--------------- 1 file changed, 3 insertions(+), 15 deletions(-) diff --git a/bigframes/core/compile/sqlglot/expressions/numeric_ops.py b/bigframes/core/compile/sqlglot/expressions/numeric_ops.py index e0ea24a470..c022356fd3 100644 --- a/bigframes/core/compile/sqlglot/expressions/numeric_ops.py +++ b/bigframes/core/compile/sqlglot/expressions/numeric_ops.py @@ -127,14 +127,7 @@ def _(expr: TypedExpr) -> sge.Expression: @register_binary_op(ops.cosine_distance_op) def _(left: TypedExpr, right: TypedExpr) -> sge.Expression: - return sge.Anonymous( - this="ML.DISTANCE", - expressions=[ - left.expr, - right.expr, - sge.Literal.string("COSINE"), - ], - ) + return sge.func("ML.DISTANCE", left.expr, right.expr, sge.Literal.string("COSINE")) @register_unary_op(ops.exp_op) @@ -307,13 +300,8 @@ def _(left: TypedExpr, right: TypedExpr) -> sge.Expression: @register_binary_op(ops.euclidean_distance_op) def _(left: TypedExpr, right: TypedExpr) -> sge.Expression: - return sge.Anonymous( - this="ML.DISTANCE", - expressions=[ - left.expr, - right.expr, - sge.Literal.string("EUCLIDEAN"), - ], + return sge.func( + "ML.DISTANCE", left.expr, right.expr, sge.Literal.string("EUCLIDEAN") )