fix: added logger.error(e) to diagnoss issues pertaining to failed ml…

…flow inits
aisingapore · Sep 11, 2024 · fa0b0ab · fa0b0ab
1 parent 6e6e507
commit fa0b0ab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/{{cookiecutter.repo_name}}/src/{{cookiecutter.src_package_name}}/general_utils.py b/{{cookiecutter.repo_name}}/src/{{cookiecutter.src_package_name}}/general_utils.py
@@ -92,8 +92,9 @@ def mlflow_init(args, run_name='train-model', setup_mlflow=False, autolog=False)
             init_success = True
             logger.info("MLflow initialisation has succeeded.")
             logger.info("UUID for MLflow run: %s", mlflow_run.info.run_id)
-        except Exception:
+        except Exception as e:
             logger.error("MLflow initialisation has failed.")
+            logger.error(e)
 
     return init_success, mlflow_run