如何从 MLflow 项目中抛出异常？

Question

我有一个引发异常的 Mlflow 项目。我使用 mlflow.run 执行该函数，但我得到 mlflow.exceptions.ExecutionException("Run (ID '<run_id>') failed")。

有什么方法可以获取在我执行 mlflow.run 时引发的异常？

或者是否可以从项目中发送带有自定义消息集的 mlflow.exceptions.ExecutionException？

Answer 1

很遗憾，目前没有。 mlflow 运行启动一个新进程，现在没有异常传递协议。一般来说，其他项目甚至不必使用相同的语言。

我能想到的一种解决方法是通过设置运行标记通过 mlflow 传递异常。例如：

try:
    ...
except Exception as ex:
    mlflow.set_tag("exception", str(ex))

How can I throw an exception from within an MLflow project?