如何在MLflow项目中引发异常？

Question

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

有什么办法可以使我在执行mlflow.run的地方遇到异常？

或者可以从项目内部发送带有自定义消息集的mlflow.exceptions.ExecutionException吗？

Answer 1

很抱歉，目前没有。 mlflow run启动一个新进程，并且目前没有用于异常传递的协议。通常，其他项目甚至不必使用相同的语言。

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

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