Increase line length to 100 (#356)

* Increase line length to 100

* black restyle after rebase

Author: crericha

.pre-commit-config.yaml

@@ -21,6 +21,8 @@ repos:
         args:
           - --target-version
           - py37
+          - --line-length
+          - '100'
         entry: black
         require_serial: true
         files: *files

examples/airflow/docker_compose_decorator.py

@@ -115,9 +115,7 @@
         environment_variables[
             "AIRFLOW_CONN_GITHUB_DEFAULT"
         ] = '{"conn_type": "github", "password": "${DEMO_GITHUB_PAT}"}'
-    docker_compose["services"][airflow_component]["environment"].update(
-        environment_variables
-    )
+    docker_compose["services"][airflow_component]["environment"].update(environment_variables)
 
 
 with open(DOCKER_COMPOSE_YAML, "w", encoding="utf-8") as fd:

examples/sushi/macros/macros.py

@@ -13,13 +13,9 @@ def add_one(evaluator: MacroEvaluator, column: str) -> exp.Expression:
 @macro()
 def incremental_by_ds(evaluator: MacroEvaluator, column: exp.Column) -> exp.Expression:
     expression = evaluator.transform(
-        exp.Between(
-            this=column, low=MacroVar(this="start_ds"), high=MacroVar(this="end_ds")
-        )
+        exp.Between(this=column, low=MacroVar(this="start_ds"), high=MacroVar(this="end_ds"))
     )
     if not isinstance(expression, exp.Expression):
-        raise MacroEvalError(
-            f"Return type is {type(expression)}, expected exp.Expression"
-        )
+        raise MacroEvalError(f"Return type is {type(expression)}, expected exp.Expression")
 
     return expression

examples/sushi/models/order_items.py

@@ -29,11 +29,7 @@
     audits=[
         (
             "not_null",
-            {
-                "columns": [
-                    to_column(c) for c in ("id", "order_id", "item_id", "quantity")
-                ]
-            },
+            {"columns": [to_column(c) for c in ("id", "order_id", "item_id", "quantity")]},
         ),
         ("assert_order_items_quantity_exceeds_threshold", {"quantity": 0}),
     ],

examples/wursthall/models/db/order_f.py

@@ -25,9 +25,7 @@
 
 @model(
     "db.order_f",
-    kind=IncrementalByTimeRangeKind(
-        time_column=TimeColumn(column="order_ds", format="%Y-%m-%d")
-    ),
+    kind=IncrementalByTimeRangeKind(time_column=TimeColumn(column="order_ds", format="%Y-%m-%d")),
     start=DATA_START_DATE_STR,
     cron="@daily",
     batch_size=200,
@@ -69,9 +67,7 @@ def execute(
 
     df_order_item_f = df_order_item_f.merge(df_item_d, how="inner", on="item_id")
     df_order_item_f["item_price"] = 1.00
-    df_order_item_f["item_total"] = (
-        df_order_item_f["item_price"] * df_order_item_f["quantity"]
-    )
+    df_order_item_f["item_total"] = df_order_item_f["item_price"] * df_order_item_f["quantity"]
     df_order_item_f = (
         df_order_item_f.groupby(["order_id", "customer_id", "order_ds"], dropna=False)
         .agg(order_total=("item_total", "sum"))

examples/wursthall/models/src/order_item_details.py

@@ -29,9 +29,7 @@ class OrderItemDetails:
 
 @model(
     "src.order_item_details",
-    kind=IncrementalByTimeRangeKind(
-        time_column=TimeColumn(column="order_ds", format="%Y-%m-%d")
-    ),
+    kind=IncrementalByTimeRangeKind(time_column=TimeColumn(column="order_ds", format="%Y-%m-%d")),
     start=DATA_START_DATE_STR,
     cron="@daily",
     batch_size=100,
@@ -99,9 +97,7 @@ def execute(
                 )
             ):
                 item_id = str(
-                    df_menu_items.iloc[[random.choice(range(num_menu_items))]][
-                        "item_id"
-                    ].values[0]
+                    df_menu_items.iloc[[random.choice(range(num_menu_items))]]["item_id"].values[0]
                 )
                 quantity = np.random.choice(range(1, 4), p=[0.8, 0.1, 0.1])
                 results.append(

sqlmesh/__init__.py

@@ -96,9 +96,7 @@ class CustomFormatter(logging.Formatter):
     red = "\x1b[31;20m"
     bold_red = "\x1b[31;1m"
     reset = "\x1b[0m"
-    log_format = (
-        "%(asctime)s - %(name)s - %(levelname)s - %(message)s (%(filename)s:%(lineno)d)"
-    )
+    log_format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s (%(filename)s:%(lineno)d)"
 
     FORMATS = {
         logging.DEBUG: grey + log_format + reset,

sqlmesh/cli/main.py

@@ -219,9 +219,7 @@ def diff(ctx: click.Context, environment: t.Optional[str] = None) -> None:
 )
 @click.pass_context
 @error_handler
-def plan(
-    ctx: click.Context, environment: t.Optional[str] = None, **kwargs: t.Any
-) -> None:
+def plan(ctx: click.Context, environment: t.Optional[str] = None, **kwargs: t.Any) -> None:
     """Plan a migration of the current context's models with the given environment."""
     context = ctx.obj
     restate_models = kwargs.pop("restate_model", None)
@@ -238,9 +236,7 @@ def plan(
 )
 @click.pass_context
 @error_handler
-def run(
-    ctx: click.Context, environment: t.Optional[str] = None, **kwargs: t.Any
-) -> None:
+def run(ctx: click.Context, environment: t.Optional[str] = None, **kwargs: t.Any) -> None:
     """Evaluates the DAG of models using the built-in scheduler."""
     context = ctx.obj
     context.run(**kwargs)

sqlmesh/core/audit/definition.py

@@ -53,9 +53,7 @@ class Audit(AuditMeta, frozen=True):
     """
 
     query: t.Union[exp.Subqueryable, d.Jinja]
-    expressions_: t.Optional[t.List[exp.Expression]] = Field(
-        default=None, alias="expressions"
-    )
+    expressions_: t.Optional[t.List[exp.Expression]] = Field(default=None, alias="expressions")
 
     _path: t.Optional[pathlib.Path] = None
 
@@ -77,9 +75,7 @@ def load(
             dialect: The default dialect if no audit dialect is configured.
         """
         if len(expressions) < 2:
-            _raise_config_error(
-                "Incomplete audit definition, missing AUDIT or QUERY", path
-            )
+            _raise_config_error("Incomplete audit definition, missing AUDIT or QUERY", path)
 
         meta, *statements, query = expressions
 
@@ -92,9 +88,7 @@ def load(
 
         provided_meta_fields = {p.name for p in meta.expressions}
 
-        missing_required_fields = AuditMeta.missing_required_fields(
-            provided_meta_fields
-        )
+        missing_required_fields = AuditMeta.missing_required_fields(provided_meta_fields)
         if missing_required_fields:
             _raise_config_error(
                 f"Missing required fields {missing_required_fields} in the audit definition",
@@ -121,11 +115,7 @@ def load(
                 **{
                     "dialect": dialect or "",
                     **AuditMeta(
-                        **{
-                            prop.name: prop.args.get("value")
-                            for prop in meta.expressions
-                            if prop
-                        },
+                        **{prop.name: prop.args.get("value") for prop in meta.expressions if prop},
                     ).dict(),
                 },
             )

sqlmesh/core/config/base.py

@@ -65,16 +65,12 @@ def update_field(
         combined = old.copy()
         for key, value in new.items():
             if not isinstance(value, list):
-                raise ConfigError(
-                    "KEY_EXTEND behavior requires list values in dictionary."
-                )
+                raise ConfigError("KEY_EXTEND behavior requires list values in dictionary.")
 
             old_value = combined.get(key)
             if old_value:
                 if not isinstance(old_value, list):
-                    raise ConfigError(
-                        "KEY_EXTEND behavior requires list values in dictionary."
-                    )
+                    raise ConfigError("KEY_EXTEND behavior requires list values in dictionary.")
 
                 combined[key] = old_value + value
             else: