From 7c45dbb407dc81e71fbdbba44d5205d14203a893 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 10 Sep 2026 09:23:31 -0300 Subject: [PATCH] QTZPOC-20 QTZPOC-20: Apply minor formatting adjustments to improve code readability. --- laborious/workflows/import_model.py | 4 +++- tests/laborious/utils/bundle/test_reader.py | 4 +++- tests/laborious/utils/repository/test_model_repository.py | 8 ++++++-- 3 files changed, 12 insertions(+), 4 deletions(-) diff --git a/laborious/workflows/import_model.py b/laborious/workflows/import_model.py index 0d8ea0d..b85df35 100644 --- a/laborious/workflows/import_model.py +++ b/laborious/workflows/import_model.py @@ -444,7 +444,9 @@ class ImportModel: retry_policy=retry_policy, start_to_close_timeout=CLEANUP_TIMEOUT, ) - except BaseException as error: # NOSONAR - cleanup never replaces the failure that brought us here + except ( + BaseException + ) as error: # NOSONAR - cleanup never replaces the failure that brought us here if is_cancelled_exception(error): raise return {'cleaned': False} diff --git a/tests/laborious/utils/bundle/test_reader.py b/tests/laborious/utils/bundle/test_reader.py index da22105..5db00ec 100644 --- a/tests/laborious/utils/bundle/test_reader.py +++ b/tests/laborious/utils/bundle/test_reader.py @@ -463,7 +463,9 @@ def test_extract_member_rechecks_containment_on_the_resolved_path(work_dir, read assert error.value.gate == 5 -def test_extract_member_rejects_a_stream_larger_than_its_declared_size(work_dir, reader, monkeypatch): +def test_extract_member_rejects_a_stream_larger_than_its_declared_size( + work_dir, reader, monkeypatch +): """A decompressed stream longer than the entry's declared `file_size` is rejected mid-write, instead of trusting the central-directory metadata gate 4 already checked in aggregate. diff --git a/tests/laborious/utils/repository/test_model_repository.py b/tests/laborious/utils/repository/test_model_repository.py index 883da62..2014bdd 100644 --- a/tests/laborious/utils/repository/test_model_repository.py +++ b/tests/laborious/utils/repository/test_model_repository.py @@ -667,7 +667,9 @@ async def test_download_model_resolves_run_id_once_non_wrapper(mlflow_repository await mlflow_repository.download_model('test_model', {}, 'predict', 'sklearn', False) - mlflow_repository.get_model_run_id.assert_called_once_with(model_name='test_model', stage='Production') + mlflow_repository.get_model_run_id.assert_called_once_with( + model_name='test_model', stage='Production' + ) mlflow_repository.download_artifact_code_dir.assert_called_once_with( 'test_model', 'prediction_model', 'sklearn', {}, run_id='run-123' ) @@ -685,7 +687,9 @@ async def test_download_model_resolves_run_id_once_wrapper(mlflow, mlflow_reposi await mlflow_repository.download_model('test_model', {}, 'predict', 'pyfunc', True) - mlflow_repository.get_model_run_id.assert_called_once_with(model_name='test_model', stage='Production') + mlflow_repository.get_model_run_id.assert_called_once_with( + model_name='test_model', stage='Production' + ) mlflow_repository.download_artifact_code_dir.assert_called_once_with( 'test_model', 'prediction_model', 'pyfunc', {}, run_id='run-123' )