Update generative-proof-of-concept-CPU-preprocessing-in-memory.py

Add run object...

Author: david-thrower

generative-proof-of-concept-CPU-preprocessing-in-memory.py

@@ -172,7 +172,7 @@ def objective(trial: optuna.Trial) -> float:
 
     tags = {"phase": "poc", "script": os.path.basename(__file__), "trial_number": str(trial.number), "Start_time": str(trial_start_time)}
 
-    with mlflow.start_run(run_name=run_name, tags=tags):
+    with mlflow.start_run(run_name=run_name, tags=tags) as run:
         # Log the hyperparameters
         mlflow.log_params(params)