ray.air.integrations.mlflow.MLflowLoggerCallback.set_state

MLflowLoggerCallback.set_state(state: Dict)

Set the state of the callback.

This method should be implemented by subclasses to restore the callback’s state based on the given dict state.

This is used automatically by Tune to restore checkpoint callback state on Tune experiment restoration.

See get_state() for an example implementation.

Parameters:

state – State of the callback.