chore: adjusts type hints to account for complications with mypy part 2

Author: chalmerlowe

packages/bigframes/bigframes/core/compile/ibis_compiler/ibis_compiler.py

@@ -49,8 +49,8 @@ def compile_sql(request: configs.CompileRequest) -> configs.CompileResult:
         # Can only pullup slice if we are doing ORDER BY in outermost SELECT
         # Need to do this before replacing unsupported ops, as that will rewrite slice ops
         result_node = rewrites.pull_up_limits(result_node)
-    result_node = _replace_unsupported_ops(result_node)
-    result_node = result_node.bottom_up(rewrites.simplify_join)
+    result_node = cast(nodes.ResultNode, _replace_unsupported_ops(result_node))
+    result_node = cast(nodes.ResultNode, result_node.bottom_up(rewrites.simplify_join))
     # prune before pulling up order to avoid unnnecessary row_number() ops
     result_node = cast(nodes.ResultNode, rewrites.column_pruning(result_node))
     result_node = rewrites.defer_order(

packages/bigframes/bigframes/core/compile/sqlglot/compiler.py

@@ -53,8 +53,8 @@ def compile_sql(request: configs.CompileRequest) -> configs.CompileResult:
         # Can only pullup slice if we are doing ORDER BY in outermost SELECT
         # Need to do this before replacing unsupported ops, as that will rewrite slice ops
         result_node = rewrite.pull_up_limits(result_node)
-    result_node = _replace_unsupported_ops(result_node)
-    result_node = result_node.bottom_up(rewrite.simplify_join)
+    result_node = typing.cast(nodes.ResultNode, _replace_unsupported_ops(result_node))
+    result_node = typing.cast(nodes.ResultNode, result_node.bottom_up(rewrite.simplify_join))
     # prune before pulling up order to avoid unnnecessary row_number() ops
     result_node = typing.cast(nodes.ResultNode, rewrite.column_pruning(result_node))
     result_node = rewrite.defer_order(

packages/bigframes/bigframes/series.py

@@ -2308,9 +2308,10 @@ def to_json(
             )
         else:
             pd_series = self.to_pandas(allow_large_results=allow_large_results)
+            # Pandas Series.to_json only supports a subset of orients, but bigframes Series.to_json allows all of them.
             return pd_series.to_json(
                 path_or_buf=path_or_buf,
-                orient=orient,
+                orient=orient,  # type: ignore[arg-type]
                 lines=lines,
                 index=index,  # type: ignore
             )