def METHOD_NAME(self, logs: Optional[Dict[Text, Any]] = None) -> None:
    """Closes the progress bar after training.
    Args:
        logs: The training metrics.
    """
    self.progress_bar.close()