fix mypy

Author: sycai

bigframes/core/compile/compiled.py

@@ -548,9 +548,11 @@ def _ibis_window_from_spec(self, window_spec: WindowSpec):
                 window_spec.ordering,
             )
         else:
-            order_by = _convert_range_ordering_to_table_value(
-                self._column_names, window_spec.ordering[0]
-            )
+            order_by = [
+                _convert_range_ordering_to_table_value(
+                    self._column_names, window_spec.ordering[0]
+                )
+            ]
 
         window = bigframes_vendored.ibis.window(order_by=order_by, group_by=group_by)
         if window_spec.bounds is not None:
@@ -611,8 +613,8 @@ def _convert_range_ordering_to_table_value(
     )
 
     if ordering_column.direction.is_ascending:
-        return bigframes_vendored.ibis.asc(expr)
-    return bigframes_vendored.ibis.desc(expr)
+        return bigframes_vendored.ibis.asc(expr)  # type: ignore
+    return bigframes_vendored.ibis.desc(expr)  # type: ignore
 
 
 def _string_cast_join_cond(

bigframes/core/compile/polars/compiler.py

@@ -16,7 +16,7 @@
 import dataclasses
 import functools
 import itertools
-from typing import cast, Optional, Sequence, Tuple, TYPE_CHECKING, Union
+from typing import cast, Optional, Sequence, Tuple, TYPE_CHECKING
 
 import bigframes.core
 from bigframes.core import window_spec
@@ -360,6 +360,7 @@ def compile_window(self, node: nodes.WindowOpNode):
             return df.with_columns([agg_expr])
 
         else:  # row-bounded window
+            assert isinstance(window.bounds, window_spec.RowsWindowBounds)
             # Polars API semi-bounded, and any grouped rolling window challenging
             # https://github.com/pola-rs/polars/issues/4799
             # https://github.com/pola-rs/polars/issues/8976
@@ -383,9 +384,7 @@ def compile_window(self, node: nodes.WindowOpNode):
             return pl.concat([df, results], how="horizontal")
 
 
-def _get_period(
-    bounds: Union[window_spec.RowsWindowBounds, window_spec.RangeWindowBounds]
-) -> Optional[int]:
+def _get_period(bounds: window_spec.RowsWindowBounds) -> Optional[int]:
     """Returns None if the boundary is infinite."""
     if bounds.start is None or bounds.end is None:
         return None