Skip to content

Commit a8fcc2f

Browse files
committed
linting and style changes
1 parent dc08ba3 commit a8fcc2f

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

sqlmesh/core/engine_adapter/bigquery.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1108,7 +1108,7 @@ def _execute(
11081108

11091109
# Create job config with reservation support
11101110
job_config = QueryJobConfig(**self._job_params, connection_properties=connection_properties)
1111-
1111+
11121112
# Set reservation directly on the job_config object if specified
11131113
reservation_id = self._extra_config.get("reservation_id")
11141114
if reservation_id:

tests/core/test_connection_config.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1050,12 +1050,18 @@ def test_bigquery(make_config):
10501050
check_import=False,
10511051
)
10521052
assert isinstance(config_with_reservation, BigQueryConnectionConfig)
1053-
assert config_with_reservation.reservation_id == "projects/my-project/locations/us-central1/reservations/my-reservation"
1054-
1053+
assert (
1054+
config_with_reservation.reservation_id
1055+
== "projects/my-project/locations/us-central1/reservations/my-reservation"
1056+
)
1057+
10551058
# Test that reservation_id is included in _extra_engine_config
10561059
extra_config = config_with_reservation._extra_engine_config
10571060
assert "reservation_id" in extra_config
1058-
assert extra_config["reservation_id"] == "projects/my-project/locations/us-central1/reservations/my-reservation"
1061+
assert (
1062+
extra_config["reservation_id"]
1063+
== "projects/my-project/locations/us-central1/reservations/my-reservation"
1064+
)
10591065

10601066
with pytest.raises(ConfigError, match="you must also specify the `project` field"):
10611067
make_config(type="bigquery", execution_project="execution_project", check_import=False)

0 commit comments

Comments
 (0)