format file

Author: shuoweil

packages/bigframes/bigframes/core/compile/sqlglot/sql/base.py

@@ -49,11 +49,9 @@
 
 def to_sql(expr: sge.Expression) -> str:
     """Generate SQL string from the given expression."""
+
     def _flatten_null_casts(node: sge.Expression) -> sge.Expression:
-        if (
-            isinstance(node, (sge.Cast, sge.TryCast))
-            and str(node.to).upper() == "NULL"
-        ):
+        if isinstance(node, (sge.Cast, sge.TryCast)) and str(node.to).upper() == "NULL":
             return sge.Null()
         return node
 

packages/bigframes/tests/unit/core/compile/sqlglot/sql/test_base.py

@@ -181,5 +181,6 @@ def test_cast_to_null_type_returns_flat_null():
 
 def test_nested_cast_to_null_type_is_flattened():
     import bigframes_vendored.sqlglot.expressions as sge
+
     nested = sge.Cast(this=sge.Cast(this=sge.Null(), to="NULL"), to="INT64")
     assert sql.to_sql(nested) == "CAST(NULL AS INT64)"