gitea_admin/sientia-dataops-laborious-legacy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

QTZPOC-20

Browse Source 
QTZPOC-20: Consolidate written_code_dir assignment into a single line in sklearn flavor retrain E2E test.
This commit is contained in:
Vitor Pimentel dos Santos
2026-09-10 11:50:51 -03:00
parent 512a49f78f
commit 4cccc45c34
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/laborious/utils/repository/test_sklearn_flavor_retrain_e2e.py
View File

@@ -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': {}}) await repository.log_model(model_data, 'sklearn', 'prediction_model', {'metadata': {}})
run_id = run.info.run_id run_id = run.info.run_id
written_code_dir = Path( written_code_dir = Path(repository.client.download_artifacts(run_id, 'prediction_model/code'))
repository.client.download_artifacts(run_id, 'prediction_model/code')
)
assert (written_code_dir / f'{CODE_MODULE_NAME}.py').is_file() 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. # The class must resolve from the artifact's written code/, not from anything left in memory.