  
              images[side] = get_image_paths(image_dir)
              if not images[side]:
                  logger.error("Error: '%s' contains no images", image_dir)
                  exit(1)
  
          logger.info("Model A Directory: %s", self._args.input_a)
          logger.info("Model B Directory: %s", self._args.input_b)
          logger.debug("Got image paths: %s", [(key, str(len(val)) + " images")
                                               for key, val in images.items()])
 
</s> remove                 exit(1)
 </s> add                 sys.exit(1) </s> remove             trainer.train_one_step(viewer, timelapse)
 </s> add             trainer.train_one_step(viewer, timelapse, self._args.colab) </s> remove                 self.__print_loss(loss)
 </s> add                 self.__print_loss(loss, is_colab) </s> remove                 self.__print_loss(self._pingpong.loss)
 </s> add                 self.__print_loss(self._pingpong.loss, is_colab) </s> remove                     "Terminate" if self._args.redirect_gui else "ENTER")
        if not self._args.redirect_gui:
 </s> add                     "Stop" if self._args.redirect_gui or self._args.colab else "ENTER")
        if not self._args.redirect_gui and not self._args.colab:
                sys.exit(1)
  
              images[side] = get_image_paths(image_dir)
              if not images[side]:
                  logger.error("Error: '%s' contains no images", image_dir)
                sys.exit(1)
  
          logger.info("Model A Directory: %s", self._args.input_a)
          logger.info("Model B Directory: %s", self._args.input_b)
          logger.debug("Got image paths: %s", [(key, str(len(val)) + " images")
                                               for key, val in images.items()])
keep keep keep keep replace keep keep keep keep keep
 lib,cli: Add suppressed colab flag 
scripts/train.py