fix annotations

Author: tswast

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

@@ -94,7 +94,7 @@ def sqrt_op_impl(
 @polars_compiler.register_op(numeric_ops.IsNanOp)
 def is_nan_op_impl(
     compiler: polars_compiler.PolarsExpressionCompiler,
-    op: numeric_ops.SqrtOp,  # type: ignore
+    op: numeric_ops.IsNanOp,  # type: ignore
     input: pl.Expr,
 ) -> pl.Expr:
     return input.is_nan()
@@ -103,7 +103,7 @@ def is_nan_op_impl(
 @polars_compiler.register_op(numeric_ops.IsFiniteOp)
 def is_finite_op_impl(
     compiler: polars_compiler.PolarsExpressionCompiler,
-    op: numeric_ops.SqrtOp,  # type: ignore
+    op: numeric_ops.IsFiniteOp,  # type: ignore
     input: pl.Expr,
 ) -> pl.Expr:
     return input.is_finite()

bigframes/operations/numeric_ops.py

@@ -350,7 +350,7 @@ def output_type(self, *input_types: dtypes.ExpressionType) -> dtypes.ExpressionT
 unsafe_pow_op = UnsafePowOp()
 
 IsNanOp = base_ops.create_unary_op(
-    name="isnanornull",
+    name="isnan",
     type_signature=op_typing.FixedOutputType(
         dtypes.is_numeric, dtypes.BOOL_DTYPE, "numeric"
     ),