From bc8e8d4ec3ba20b0b94ba3266e2be994754a85d0 Mon Sep 17 00:00:00 2001 From: Alex Streed Date: Fri, 7 Feb 2025 15:46:15 -0600 Subject: [PATCH] Fix formatting --- src/prefect/deployments/runner.py | 4 ++-- tests/runner/test_runner.py | 12 +++++++----- 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/src/prefect/deployments/runner.py b/src/prefect/deployments/runner.py index 180c33013ded..de3079d6fc73 100644 --- a/src/prefect/deployments/runner.py +++ b/src/prefect/deployments/runner.py @@ -348,7 +348,7 @@ async def _create( return deployment_id async def _update(self, deployment_id: UUID, client: PrefectClient): - parameter_openapi_schema=self._parameter_openapi_schema.model_dump( + parameter_openapi_schema = self._parameter_openapi_schema.model_dump( exclude_unset=True ) await client.update_deployment( @@ -359,7 +359,7 @@ async def _update(self, deployment_id: UUID, client: PrefectClient): mode="json", exclude_unset=True, exclude={"storage", "name", "flow_name", "triggers"}, - ) + ), ), ) diff --git a/tests/runner/test_runner.py b/tests/runner/test_runner.py index a1fcc155567a..f81509cc9ba3 100644 --- a/tests/runner/test_runner.py +++ b/tests/runner/test_runner.py @@ -426,7 +426,9 @@ async def test_add_flows_to_runner(self, prefect_client: PrefectClient): assert deployment_2.name == "test_runner" assert deployment_2.schedules[0].schedule.cron == "* * * * *" - async def test_add_flow_to_runner_always_updates_openapi_schema(self, prefect_client: PrefectClient): + async def test_add_flow_to_runner_always_updates_openapi_schema( + self, prefect_client: PrefectClient + ): """Runner.add should create a deployment for the flow passed to it""" runner = Runner() @@ -438,8 +440,8 @@ def one(num: int): deployment = await prefect_client.read_deployment(deployment_id) assert deployment.name == "test-openapi" - assert deployment.description == 'None' - assert set(deployment.parameter_openapi_schema['properties'].keys()) == {'num'} + assert deployment.description == "None" + assert set(deployment.parameter_openapi_schema["properties"].keys()) == {"num"} @flow(name="one") def two(num: int): @@ -451,7 +453,7 @@ def two(num: int): assert deployment.name == "test-openapi" assert deployment.description == "description now" - assert set(deployment.parameter_openapi_schema['properties'].keys()) == {'num'} + assert set(deployment.parameter_openapi_schema["properties"].keys()) == {"num"} @flow(name="one") def three(name: str): @@ -462,7 +464,7 @@ def three(name: str): assert deployment.name == "test-openapi" assert deployment.description is None - assert set(deployment.parameter_openapi_schema['properties'].keys()) == {'name'} + assert set(deployment.parameter_openapi_schema["properties"].keys()) == {"name"} @pytest.mark.parametrize( "kwargs",