Skip to content

Commit 7ed9029

Browse files
committed
revert sqrt change
1 parent f36cce2 commit 7ed9029

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

bigframes/core/compile/polars/operations/numeric_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,4 +88,4 @@ def sqrt_op_impl(
8888
) -> pl.Expr:
8989
import polars as pl
9090

91-
return pl.when(input <= 0).then(float("nan")).otherwise(input.sqrt())
91+
return pl.when(input < 0).then(float("nan")).otherwise(input.sqrt())

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def _(expr: TypedExpr) -> sge.Expression:
207207
return sge.Case(
208208
ifs=[
209209
sge.If(
210-
this=expr.expr <= sge.convert(0),
210+
this=expr.expr < sge.convert(0),
211211
true=constants._NAN,
212212
)
213213
],

0 commit comments

Comments
 (0)