ray.air.integrations.mlflow.MLflowLoggerCallback.on_step_end

MLflowLoggerCallback.on_step_end(iteration: int, trials: List[Trial], **info)

Called at the end of each tuning loop step.

The iteration counter is increased before this hook is called.

Parameters:

• iteration – Number of iterations of the tuning loop.

• trials – List of trials.

• **info – Kwargs dict for forward compatibility.