zendag.mlflow_utils

zendag.mlflow_utils.mlflow_run(project_name='DefaultProject', stage_dir_fn=<function default_stage_dir_fn>, configs_dir_fn=<function default_configs_dir_fn>)

Decorator to wrap a function execution within nested MLflow runs.

• Establishes a parent run using MLFLOW_RUN_ID (if set via .pipeline_id).

• Creates a nested child run named after the DVC stage or function name.

• Logs Hydra configuration parameters from the corresponding composed config file.

• Logs the composed config file itself as an artifact.

• Logs the Hydra run log file (run.log) as an artifact upon completion or failure.

• Handles exceptions and ensures logs are captured if possible.