From 4cccc45c34cf887b346fe4101f272dafbe5b151e Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 10 Sep 2026 11:50:51 -0300 Subject: [PATCH] QTZPOC-20 QTZPOC-20: Consolidate written_code_dir assignment into a single line in sklearn flavor retrain E2E test. --- .../utils/repository/test_sklearn_flavor_retrain_e2e.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/laborious/utils/repository/test_sklearn_flavor_retrain_e2e.py b/tests/laborious/utils/repository/test_sklearn_flavor_retrain_e2e.py index f1b77fe..965b61e 100644 --- a/tests/laborious/utils/repository/test_sklearn_flavor_retrain_e2e.py +++ b/tests/laborious/utils/repository/test_sklearn_flavor_retrain_e2e.py @@ -87,9 +87,7 @@ async def test_retrain_sklearn_model_preserves_code_dir_and_reloads(tmp_path): await repository.log_model(model_data, 'sklearn', 'prediction_model', {'metadata': {}}) run_id = run.info.run_id - written_code_dir = Path( - repository.client.download_artifacts(run_id, 'prediction_model/code') - ) + written_code_dir = Path(repository.client.download_artifacts(run_id, 'prediction_model/code')) assert (written_code_dir / f'{CODE_MODULE_NAME}.py').is_file() # The class must resolve from the artifact's written code/, not from anything left in memory.