reformat

Author: TrevorBergeron

packages/bigframes/bigframes/session/_io/bigquery/__init__.py

@@ -528,7 +528,9 @@ def to_query(
     if columns:
         # We only reduce the selection if columns is set, but we always
         # want to make sure index_cols is also included.
-        select_clause = "SELECT " + ", ".join(f"`_bf_source`.`{column}`" for column in columns)
+        select_clause = "SELECT " + ", ".join(
+            f"`_bf_source`.`{column}`" for column in columns
+        )
     else:
         select_clause = "SELECT *"
 

packages/bigframes/third_party/bigframes_vendored/sqlglot/parser.py

@@ -290,11 +290,11 @@ class Parser(metaclass=_Parser):
         "RIGHTPAD": lambda args: build_pad(args, is_left=False),
         "RPAD": lambda args: build_pad(args, is_left=False),
         "RTRIM": lambda args: build_trim(args, is_left=False),
-        "SCOPE_RESOLUTION": lambda args: exp.ScopeResolution(
-            expression=seq_get(args, 0)
-        )
-        if len(args) != 2
-        else exp.ScopeResolution(this=seq_get(args, 0), expression=seq_get(args, 1)),
+        "SCOPE_RESOLUTION": lambda args: (
+            exp.ScopeResolution(expression=seq_get(args, 0))
+            if len(args) != 2
+            else exp.ScopeResolution(this=seq_get(args, 0), expression=seq_get(args, 1))
+        ),
         "STRPOS": exp.StrPosition.from_arg_list,
         "CHARINDEX": lambda args: build_locate_strposition(args),
         "INSTR": exp.StrPosition.from_arg_list,
@@ -943,7 +943,9 @@ class Parser(metaclass=_Parser):
     }
 
     UNARY_PARSERS = {
-        TokenType.PLUS: lambda self: self._parse_unary(),  # Unary + is handled as a no-op
+        TokenType.PLUS: lambda self: (
+            self._parse_unary()
+        ),  # Unary + is handled as a no-op
         TokenType.NOT: lambda self: self.expression(
             exp.Not, this=self._parse_equality()
         ),
@@ -1246,12 +1248,14 @@ class Parser(metaclass=_Parser):
             exp.NotNullColumnConstraint, allow_null=True
         ),
         "ON": lambda self: (
-            self._match(TokenType.UPDATE)
-            and self.expression(
-                exp.OnUpdateColumnConstraint, this=self._parse_function()
+            (
+                self._match(TokenType.UPDATE)
+                and self.expression(
+                    exp.OnUpdateColumnConstraint, this=self._parse_function()
+                )
             )
-        )
-        or self.expression(exp.OnProperty, this=self._parse_id_var()),
+            or self.expression(exp.OnProperty, this=self._parse_id_var())
+        ),
         "PATH": lambda self: self.expression(
             exp.PathColumnConstraint, this=self._parse_string()
         ),
@@ -3885,8 +3889,9 @@ def _parse_hint_body(self) -> t.Optional[exp.Hint]:
         try:
             for hint in iter(
                 lambda: self._parse_csv(
-                    lambda: self._parse_hint_function_call()
-                    or self._parse_var(upper=True),
+                    lambda: (
+                        self._parse_hint_function_call() or self._parse_var(upper=True)
+                    ),
                 ),
                 [],
             ):
@@ -4305,8 +4310,9 @@ def _parse_table_hints(self) -> t.Optional[t.List[exp.Expression]]:
                 self.expression(
                     exp.WithTableHint,
                     expressions=self._parse_csv(
-                        lambda: self._parse_function()
-                        or self._parse_var(any_token=True)
+                        lambda: (
+                            self._parse_function() or self._parse_var(any_token=True)
+                        )
                     ),
                 )
             )
@@ -4485,7 +4491,6 @@ def _parse_table(
         if self.dialect.ALIAS_POST_TABLESAMPLE:
             this.set("sample", self._parse_table_sample())
 
-
         if self._match(TokenType.INDEXED_BY):
             this.set("indexed", self._parse_table_parts())
         elif self._match_text_seq("NOT", "INDEXED"):
@@ -4938,11 +4943,13 @@ def _parse_group(self, skip_group_by_token: bool = False) -> t.Optional[exp.Grou
 
             elements["expressions"].extend(
                 self._parse_csv(
-                    lambda: None
-                    if self._match_set(
-                        (TokenType.CUBE, TokenType.ROLLUP), advance=False
+                    lambda: (
+                        None
+                        if self._match_set(
+                            (TokenType.CUBE, TokenType.ROLLUP), advance=False
+                        )
+                        else self._parse_disjunction()
                     )
-                    else self._parse_disjunction()
                 )
             )
 
@@ -6228,9 +6235,9 @@ def _parse_column_ops(
             # https://cloud.google.com/bigquery/docs/reference/standard-sql/functions-reference#function_call_rules
             if isinstance(field, (exp.Func, exp.Window)) and this:
                 this = this.transform(
-                    lambda n: n.to_dot(include_dots=False)
-                    if isinstance(n, exp.Column)
-                    else n
+                    lambda n: (
+                        n.to_dot(include_dots=False) if isinstance(n, exp.Column) else n
+                    )
                 )
 
             if op: