code: def __init__(self, log_models, metrics_logger):
      self.log_models = log_models
      self._logger = metrics_logger
      self.current_epoch = 0
  
file_path: mlflow/mlflow/gluon/_autolog.py
imports: from mxnet.gluon.contrib.estimator import EpochEnd, TrainBegin, TrainEnd
  from mxnet.gluon.nn import HybridSequential
  import mlflow
  from mlflow.utils.autologging_utils import ExceptionSafeClass, get_autologging_config
label_desc: Use image node_id 1 to create a new __MLflowGluonCallback object from inherited base classes: EpochEnd, TrainEnd, TrainBegin with example: obj = __MLflowGluonCallback(log_models, metrics_logger)