def METHOD_NAME(self, batch: Any, step_out: Any) -> None:
    # language=rst
    """
    Create any plots and logs for a step given the input batch and step output.
    :param batch: The current batch. This could be anything as long as the subclass
        agrees upon the format in some way.
    :param step_out: The output from the ``step_()`` method.
    """
    raise NotImplementedError("You need to provide a plots method.")