Skip to content

Commit f21c523

Browse files
committed
Refactor: Use sge.func for ML.DISTANCE
1 parent f73fb98 commit f21c523

File tree

1 file changed

+3
-15
lines changed

1 file changed

+3
-15
lines changed

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

Lines changed: 3 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -127,14 +127,7 @@ def _(expr: TypedExpr) -> sge.Expression:
127127

128128
@register_binary_op(ops.cosine_distance_op)
129129
def _(left: TypedExpr, right: TypedExpr) -> sge.Expression:
130-
return sge.Anonymous(
131-
this="ML.DISTANCE",
132-
expressions=[
133-
left.expr,
134-
right.expr,
135-
sge.Literal.string("COSINE"),
136-
],
137-
)
130+
return sge.func("ML.DISTANCE", left.expr, right.expr, sge.Literal.string("COSINE"))
138131

139132

140133
@register_unary_op(ops.exp_op)
@@ -307,13 +300,8 @@ def _(left: TypedExpr, right: TypedExpr) -> sge.Expression:
307300

308301
@register_binary_op(ops.euclidean_distance_op)
309302
def _(left: TypedExpr, right: TypedExpr) -> sge.Expression:
310-
return sge.Anonymous(
311-
this="ML.DISTANCE",
312-
expressions=[
313-
left.expr,
314-
right.expr,
315-
sge.Literal.string("EUCLIDEAN"),
316-
],
303+
return sge.func(
304+
"ML.DISTANCE", left.expr, right.expr, sge.Literal.string("EUCLIDEAN")
317305
)
318306

319307

0 commit comments

Comments
 (0)