0

I have an Mlflow project that raises an exception. I execute that function using mlflow.run, but I get mlflow.exceptions.ExecutionException("Run (ID '<run_id>') failed").

Is there any way I could get the exception that is being raised where I am executing mlflow.run?

Or is it possible to send an mlflow.exceptions.ExecutionException with custom message set from within the project?

Aditya Khadse
  • 3
  • 1

1 Answers1

1

Unfortunately not at the moment. mlflow run starts a new process and there is no protocol for exception passing right now. In general the other project does not even have to be in the same language.

One workaround I can think of is to pass the exception via mlflow by setting run tag. E.g.:

try:
    ...
except Exception as ex:
    mlflow.set_tag("exception", str(ex))
Harshal Parekh
  • 5,918
  • 4
  • 21
  • 43
Tomas
  • 189
  • 1