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.