def register_global_step(mr):
    """
    Add global step to ModelRegistry.

    Parameters
    ----------
    mr : ModelRegistry
    """

    def callback(**kwargs):
        mr.global_step_ = kwargs["global_step"]

    return callback