Mlflow plugin doesn't pass down run/success

[ssabdb]

Current behavior

The mlflow plugin doesn't log the graph outcome, which leads to mlflow UI appearing to report a run was a success.

hamilton/hamilton/plugins/h_mlflow.py

Line 332 in 45feeaf

mlflow.end_run()

Stack Traces

(If applicable)

Screenshots

Steps to replicate behavior

1. Use hamilton mlflow integration
2. Create and run dag with exception
3. Observe logged run_id displays as succeeded despite failure

Library & System Information

python3.11
mlflow==3.5.1
sf-hamilton[pandera,tqdm]>=1.88.0

Expected behavior

Mlflow should display the run has failed rather than succeeded.

[skrawcz]

Thanks @ssabdb, did you have a fix in mind by chance? :) or?