fix mypy

Author: chelsea-lin

bigframes/core/compile/sqlglot/aggregations/op_registration.py

@@ -44,7 +44,7 @@ def arg_checker(*args, **kwargs):
             key = op if isinstance(op, type) else type(op)
             if key in self._registered_ops:
                 raise ValueError(f"{key} is already registered")
-            self._registered_ops[key] = item
+            self._registered_ops[str(key)] = item
             return arg_checker
 
         return decorator
@@ -57,5 +57,5 @@ def __getitem__(self, op: str | agg_ops.WindowOp) -> CompilationFunc:
                 key = typing.cast(str, op.name)
                 if key in self._registered_ops:
                     return self._registered_ops[key]
-                return self._registered_ops[type(op)]
+                return self._registered_ops[str(type(op))]
         return self._registered_ops[op]

bigframes/core/compile/sqlglot/aggregations/unary_compiler.py

@@ -148,7 +148,7 @@ def _(
 ) -> sge.Expression:
     # TODO: Support interpolation argument
     # TODO: Support percentile_disc
-    result = sge.func("PERCENTILE_CONT", column.expr, sge.convert(op.q))
+    result: sge.Expression = sge.func("PERCENTILE_CONT", column.expr, sge.convert(op.q))
     if window is None:
         # PERCENTILE_CONT is a navigation function, not an aggregate function, so it always needs an OVER clause.
         result = sge.Window(this=result)

bigframes/operations/aggregations.py

@@ -223,10 +223,13 @@ def output_type(self, *input_types: dtypes.ExpressionType) -> dtypes.ExpressionT
 
 @dataclasses.dataclass(frozen=True)
 class QuantileOp(UnaryAggregateOp):
-    name: typing.ClassVar[str] = "quantile"
     q: float
     should_floor_result: bool = False
 
+    @property
+    def name(self):
+        return f"{int(self.q * 100)}%"
+
     @property
     def order_independent(self) -> bool:
         return True