  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  
  
  class TrainerBase():
      """ Base Trainer """
  
      def __init__(self, model, images, batch_size, configfile):
 
</s> remove     def __init__(self, model, images, batch_size):
 </s> add     def __init__(self, model, images, batch_size, configfile): </s> remove from ._base import TrainerBase as Trainer
 </s> add from ._base import TrainerBase


class Trainer(TrainerBase):
    """ Original is currently identical to Base """
    def __init__(self, *args, **kwargs):  # pylint:disable=useless-super-delegation
        super().__init__(*args, **kwargs)
 </s> remove     def __init__(self, side, images, model, use_mask, batch_size):
        logger.debug("Initializing %s: side: '%s', num_images: %s, batch_size: %s)",
                     self.__class__.__name__, side, len(images), batch_size)
 </s> add     def __init__(self, side, images, model, use_mask, batch_size, config):
        logger.debug("Initializing %s: side: '%s', num_images: %s, batch_size: %s, config: %s)",
                     self.__class__.__name__, side, len(images), batch_size, config) </s> remove     def __init__(self, model, use_mask, coverage_ratio, batchers):
 </s> add     def __init__(self, model, use_mask, coverage_ratio, preview_images, batchers):
def get_config(plugin_name, configfile=None):
    """ Return the config for the requested model """
    return Config(plugin_name, configfile=configfile).config_dict


  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  
  
def get_config(plugin_name, configfile=None):
    """ Return the config for the requested model """
    return Config(plugin_name, configfile=configfile).config_dict


  class TrainerBase():
      """ Base Trainer """
  
      def __init__(self, model, images, batch_size, configfile):
keep keep add keep keep keep keep
 Expose Augmentation Options to config 
plugins/train/trainer/_base.py