chore: update ruff version (#4138)

Author: benfdking

pyproject.toml

@@ -83,7 +83,7 @@ dev = [
     "pytest-xdist",
     "pytz",
     "redshift_connector",
-    "ruff~=0.7.0",
+    "ruff~=0.11.0",
     "snowflake-connector-python[pandas,secure-local-storage]>=3.0.2",
     "sqlalchemy-stubs",
     "trino",

sqlmesh/core/config/connection.py

@@ -279,7 +279,7 @@ def init(cursor: duckdb.DuckDBPyConnection) -> None:
                             'database with name "memory" already exists' in str(e)
                             and path_options == ":memory:"
                         )
-                        and f"""database with name "{path_options.path.replace('md:', '')}" already exists"""
+                        and f"""database with name "{path_options.path.replace("md:", "")}" already exists"""
                         not in str(e)
                     ):
                         raise e

sqlmesh/core/console.py

@@ -395,7 +395,7 @@ def _limit_model_names(self, tree: Tree, verbosity: Verbosity = Verbosity.DEFAUL
         ):
             tree.children = [
                 tree.children[0],
-                Tree(f".... {modified_length-2} more ...."),
+                Tree(f".... {modified_length - 2} more ...."),
                 tree.children[-1],
             ]
         return tree
@@ -1985,9 +1985,9 @@ def _get_snapshot_change_category(
             snapshot, plan_builder.environment_naming_info, default_catalog
         )
         response = self._prompt(
-            "\n".join([f"[{i+1}] {choice}" for i, choice in enumerate(choices.values())]),
+            "\n".join([f"[{i + 1}] {choice}" for i, choice in enumerate(choices.values())]),
             show_choices=False,
-            choices=[f"{i+1}" for i in range(len(choices))],
+            choices=[f"{i + 1}" for i in range(len(choices))],
         )
         choice = list(choices)[int(response) - 1]
         plan_builder.set_choice(snapshot, choice)
@@ -2476,7 +2476,7 @@ def show_model_difference_summary(
                 and list_length > self.INDIRECTLY_MODIFIED_DISPLAY_THRESHOLD
             ):
                 self._print(added_models[0])
-                self._print(f"- `.... {list_length-2} more ....`\n")
+                self._print(f"- `.... {list_length - 2} more ....`\n")
                 self._print(added_models[-1])
             else:
                 for snapshot in added_models:
@@ -2503,7 +2503,7 @@ def show_model_difference_summary(
                 and list_length > self.INDIRECTLY_MODIFIED_DISPLAY_THRESHOLD
             ):
                 self._print(removed_models[0])
-                self._print(f"- `.... {list_length-2} more ....`\n")
+                self._print(f"- `.... {list_length - 2} more ....`\n")
                 self._print(removed_models[-1])
             else:
                 for snapshot_table_info in removed_models:
@@ -2551,7 +2551,7 @@ def show_model_difference_summary(
                 ):
                     self._print(
                         f"- `{indirectly_modified[0].display_name(environment_naming_info, default_catalog if self.verbosity < Verbosity.VERY_VERBOSE else None, dialect=self.dialect)}`\n"
-                        f"- `.... {modified_length-2} more ....`\n"
+                        f"- `.... {modified_length - 2} more ....`\n"
                         f"- `{indirectly_modified[-1].display_name(environment_naming_info, default_catalog if self.verbosity < Verbosity.VERY_VERBOSE else None, dialect=self.dialect)}`"
                     )
                 else:
@@ -2597,7 +2597,7 @@ def _show_missing_dates(self, plan: Plan, default_catalog: t.Optional[str]) -> N
             and length > self.INDIRECTLY_MODIFIED_DISPLAY_THRESHOLD
         ):
             self._print(snapshots[0])
-            self._print(f"- `.... {length-2} more ....`\n")
+            self._print(f"- `.... {length - 2} more ....`\n")
             self._print(snapshots[-1])
         else:
             for snap in snapshots:

sqlmesh/core/test/result.py

@@ -112,8 +112,8 @@ def log_test_report(self, test_duration: float) -> None:
         # Output final report
         stream.writeln(unittest.TextTestResult.separator2)
         stream.writeln(
-            f'Ran {tests_run} {"tests" if tests_run > 1 else "test"} in {test_duration:.3f}s \n'
+            f"Ran {tests_run} {'tests' if tests_run > 1 else 'test'} in {test_duration:.3f}s \n"
         )
         stream.writeln(
-            f'{"OK" if is_success else "FAILED"}{" (" + ", ".join(infos) + ")" if infos else ""}'
+            f"{'OK' if is_success else 'FAILED'}{' (' + ', '.join(infos) + ')' if infos else ''}"
         )

sqlmesh/utils/metaprogramming.py

@@ -544,7 +544,7 @@ def format_evaluated_code_exception(
 
         error_match = re.search("^.*?Error: ", error_line)
         if error_match:
-            tb.append(f"{indent*2}  {error_line}")
+            tb.append(f"{indent * 2}  {error_line}")
             continue
 
         eval_code_match = re.search('File "<string>", line (.*), in (.*)', error_line)

tests/core/engine_adapter/integration/__init__.py

@@ -290,7 +290,7 @@ def get_table_comment(
                 WHERE
                     n.nspname = '{schema_name}'
                     AND pgc.relname = '{table_name}'
-                    AND pgc.relkind = '{'v' if table_kind == "VIEW" else 'r'}'
+                    AND pgc.relkind = '{"v" if table_kind == "VIEW" else "r"}'
                 ;
             """
         elif self.dialect in ["mysql", "snowflake"]:
@@ -362,7 +362,7 @@ def get_table_comment(
                 WHERE 
                     c.relname = '{table_name}'
                     AND n.nspname= '{schema_name}'
-                    AND c.relkind = '{'v' if table_kind == "VIEW" else 'r'}'
+                    AND c.relkind = '{"v" if table_kind == "VIEW" else "r"}'
                 ;
             """
 
@@ -405,7 +405,7 @@ def get_column_comments(
                 WHERE
                     n.nspname = '{schema_name}'
                     AND pgc.relname = '{table_name}'
-                    AND pgc.relkind = '{'v' if table_kind == "VIEW" else 'r'}'
+                    AND pgc.relkind = '{"v" if table_kind == "VIEW" else "r"}'
                 ;
             """
         elif self.dialect in ["mysql", "snowflake", "trino"]:
@@ -471,7 +471,7 @@ def get_column_comments(
                 WHERE
                     n.nspname = '{schema_name}'
                     AND c.relname = '{table_name}' 
-                    AND c.relkind = '{'v' if table_kind == "VIEW" else 'r'}'
+                    AND c.relkind = '{"v" if table_kind == "VIEW" else "r"}'
                 ;
             """
 

tests/dbt/test_transformation.py

@@ -881,9 +881,7 @@ def test_modules(sushi_test_project: Project):
     assert context.render("{{ modules.re.search('(?<=abc)def', 'abcdef').group(0) }}") == "def"
 
     # itertools
-    itertools_jinja = (
-        "{% for num in modules.itertools.accumulate([5]) %}" "{{ num }}" "{% endfor %}"
-    )
+    itertools_jinja = "{% for num in modules.itertools.accumulate([5]) %}{{ num }}{% endfor %}"
     assert context.render(itertools_jinja) == "5"
 
 
@@ -920,7 +918,7 @@ def test_column(sushi_test_project: Project):
     assert context.render("{{ api.Column }}") == "<class 'dbt.adapters.base.column.Column'>"
 
     jinja = (
-        "{% set col = api.Column('foo', 'integer') %}" "{{ col.is_integer() }} {{ col.is_string()}}"
+        "{% set col = api.Column('foo', 'integer') %}{{ col.is_integer() }} {{ col.is_string()}}"
     )
 
     assert context.render(jinja) == "True False"

tests/schedulers/airflow/test_client.py

@@ -107,7 +107,7 @@ def test_apply_plan(mocker: MockerFixture, snapshot: Snapshot):
                         "clustered_by": [],
                         "cron": "@daily",
                         "dialect": "spark",
-                        "pre_statements": ["@DEF(key, " "'value')"],
+                        "pre_statements": ["@DEF(key, 'value')"],
                         "kind": {
                             "name": "INCREMENTAL_BY_TIME_RANGE",
                             "time_column": {"column": "`ds`"},

tests/utils/test_dag.py

@@ -82,7 +82,7 @@ def test_sorted_with_cycles():
 
     expected_error_message = (
         "Last nodes added to the DAG: c\n"
-        "Possible candidates to check for circular references: b, d"
+        + "Possible candidates to check for circular references: b, d"
     )
 
     assert expected_error_message in str(ex.value)