diff --git a/src/brad/blueprint/serde.py b/src/brad/blueprint/serde.py index 2f768e5c..25abdf53 100644 --- a/src/brad/blueprint/serde.py +++ b/src/brad/blueprint/serde.py @@ -51,7 +51,7 @@ def serialize_blueprint(blueprint: Blueprint) -> bytes: def _tables_with_locations_to_proto( - table_with_locations: Tuple[Table, List[Engine]] + table_with_locations: Tuple[Table, List[Engine]], ) -> b.Table: table, locations = table_with_locations return b.Table( diff --git a/src/brad/planner/enumeration/provisioning.py b/src/brad/planner/enumeration/provisioning.py index 3c204481..7d3f99a6 100644 --- a/src/brad/planner/enumeration/provisioning.py +++ b/src/brad/planner/enumeration/provisioning.py @@ -161,7 +161,7 @@ def _load_instance_resources(file_name: str) -> Dict[str, Dict[str, int | float] def _find_min_instance( - instances: Dict[str, Dict[str, int | float]] + instances: Dict[str, Dict[str, int | float]], ) -> Tuple[str, Dict[str, int | float]]: min_type: Optional[str] = None min_resources: Optional[Dict[str, int | float]] = None diff --git a/src/brad/planner/metrics.py b/src/brad/planner/metrics.py index 58f6e2f6..a8bf5a64 100644 --- a/src/brad/planner/metrics.py +++ b/src/brad/planner/metrics.py @@ -766,7 +766,7 @@ def _fill_empty_metrics(to_fill: pd.DataFrame, guide: pd.DataFrame) -> pd.DataFr def _compute_redshift_cpu_stats( - redshift_cpu: Dict[str, float] + redshift_cpu: Dict[str, float], ) -> Tuple[float, npt.NDArray]: """ Returns the aggregated CPU value and list of all CPU values.