From 0774b1a401f9ec05651d4f9973deedcebdb5190c Mon Sep 17 00:00:00 2001 From: jj Date: Mon, 5 Feb 2024 12:12:54 +0100 Subject: [PATCH] one more thing --- model_search_upload.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/model_search_upload.py b/model_search_upload.py index edc6439..ab5b4a5 100644 --- a/model_search_upload.py +++ b/model_search_upload.py @@ -94,13 +94,12 @@ def uploat_to_mlflow(temp_dir, **context): for m in metrics: if m not in dct: continue - + print(f"Logging metric {m} {dct[m][i]}") mlflow.log_metric(key=m, value=dct[m][i]) #clean up - if 'temp_dir' in attrs: - shutil.rmtree(path=attrs["temp_dir"]) + shutil.rmtree(temp_dir) setup_task = PythonOperator(python_callable=setup, task_id="setup_connection") a_id = setup_task.output["return_value"]