                  logger.info("Adding new config item to state file: '%s': '%s'", key, val)
                  self.config[key] = val
          self.update_changed_config_items(config_changeable_items)
          logger.debug("Replacing config. Old config: %s", _CONFIG)
          _CONFIG = self.config
          if legacy_update:
              self.save()
          logger.debug("Replaced config. New config: %s", _CONFIG)
 
</s> remove 
    def update_legacy_config(self):
        """ Update legacy state config files with the new loss formating
        """
        prior = "dssim_loss"
        new = "loss_function"
        if prior in self.config.keys() and new not in self.config.keys():
            self.config[new] = "ssim" if self.config[prior] is True else "mae"
            logger.debug("Updated config from older dssim format. New config loss function: %s",
                         self.config[new])
 </s> add  </s> remove                 self.update_legacy_config()
 </s> add  </s> remove         self.funcs = self.get_loss_functions(side, predict, mask_input)
 </s> add         self.funcs = self.get_loss_functions(mask_input)
        legacy_update = self.update_legacy_config()
                  logger.info("Adding new config item to state file: '%s': '%s'", key, val)
                  self.config[key] = val
        legacy_update = self.update_legacy_config()
          self.update_changed_config_items(config_changeable_items)
          logger.debug("Replacing config. Old config: %s", _CONFIG)
          _CONFIG = self.config
          if legacy_update:
              self.save()
          logger.debug("Replaced config. New config: %s", _CONFIG)
keep add keep keep keep keep keep keep
 Post PR Fixups

- train/_config.py: PEP8 Fixes. Slight description change on coverage
- models/_base.py:
   - Remove unused variables from Loss()
   - Delete legacy config items from state file
   - Save state file on Legacy update
   - PEP8
- Remove _defaults.py for models with no config options 
plugins/train/model/_base.py