refactor: rename rewrite.operators file to rewrite.timedeltas (#1382)

Author: sycai

bigframes/core/compile/compiler.py

@@ -58,7 +58,7 @@ def compile_sql(
         # TODO: get rid of output_ids arg
         assert len(output_ids) == len(list(node.fields))
         node = set_output_names(node, output_ids)
-        node = nodes.top_down(node, rewrites.rewrite_timedelta_ops)
+        node = nodes.top_down(node, rewrites.rewrite_timedelta_expressions)
         if ordered:
             node, limit = rewrites.pullup_limit_from_slice(node)
             node = nodes.bottom_up(node, rewrites.rewrite_slice)
@@ -85,7 +85,7 @@ def compile_sql(
     def compile_peek_sql(self, node: nodes.BigFrameNode, n_rows: int) -> str:
         ids = [id.sql for id in node.ids]
         node = nodes.bottom_up(node, rewrites.rewrite_slice)
-        node = nodes.top_down(node, rewrites.rewrite_timedelta_ops)
+        node = nodes.top_down(node, rewrites.rewrite_timedelta_expressions)
         node, _ = rewrites.pull_up_order(
             node, order_root=False, ordered_joins=self.strict
         )
@@ -99,7 +99,7 @@ def compile_raw(
         str, typing.Sequence[google.cloud.bigquery.SchemaField], bf_ordering.RowOrdering
     ]:
         node = nodes.bottom_up(node, rewrites.rewrite_slice)
-        node = nodes.top_down(node, rewrites.rewrite_timedelta_ops)
+        node = nodes.top_down(node, rewrites.rewrite_timedelta_expressions)
         node, ordering = rewrites.pull_up_order(node, ordered_joins=self.strict)
         node = rewrites.column_pruning(node)
         ir = self.compile_node(node)
@@ -108,7 +108,7 @@ def compile_raw(
 
     def _preprocess(self, node: nodes.BigFrameNode):
         node = nodes.bottom_up(node, rewrites.rewrite_slice)
-        node = nodes.top_down(node, rewrites.rewrite_timedelta_ops)
+        node = nodes.top_down(node, rewrites.rewrite_timedelta_expressions)
         node, _ = rewrites.pull_up_order(
             node, order_root=False, ordered_joins=self.strict
         )

bigframes/core/rewrite/__init__.py

@@ -15,16 +15,16 @@
 from bigframes.core.rewrite.identifiers import remap_variables
 from bigframes.core.rewrite.implicit_align import try_row_join
 from bigframes.core.rewrite.legacy_align import legacy_join_as_projection
-from bigframes.core.rewrite.operators import rewrite_timedelta_ops
 from bigframes.core.rewrite.order import pull_up_order
 from bigframes.core.rewrite.pruning import column_pruning
 from bigframes.core.rewrite.slices import pullup_limit_from_slice, rewrite_slice
+from bigframes.core.rewrite.timedeltas import rewrite_timedelta_expressions
 
 __all__ = [
     "legacy_join_as_projection",
     "try_row_join",
     "rewrite_slice",
-    "rewrite_timedelta_ops",
+    "rewrite_timedelta_expressions",
     "pullup_limit_from_slice",
     "remap_variables",
     "pull_up_order",

bigframes/core/rewrite/operators.py bigframes/core/rewrite/timedeltas.pybigframes/core/rewrite/operators.py renamed to bigframes/core/rewrite/timedeltas.py

@@ -28,7 +28,7 @@ class _TypedExpr:
     dtype: dtypes.Dtype
 
 
-def rewrite_timedelta_ops(root: nodes.BigFrameNode) -> nodes.BigFrameNode:
+def rewrite_timedelta_expressions(root: nodes.BigFrameNode) -> nodes.BigFrameNode:
     """
     Rewrites expressions to properly handle timedelta values, because this type does not exist
     in the SQL world.