  from tensorflow.python.framework import (  # pylint:disable=no-name-in-module
      errors_impl as tf_errors)
  
  from lib.training import PreviewDataGenerator, TrainingDataGenerator
  from lib.training.generator import BatchType, DataGenerator
  from lib.utils import FaceswapError, get_backend, get_folder, get_image_paths, get_tf_version
  from plugins.train._config import Config
  
 
</s> remove from lib.training.generator import BatchType, ConfigType, DataGenerator
 </s> add from lib.training.generator import BatchType, DataGenerator </s> remove from typing import cast, Dict, Generator, List, Tuple, TYPE_CHECKING, Union
 </s> add from typing import cast, Dict, Generator, List, Tuple, TYPE_CHECKING
from lib.image import hex_to_rgb
  from tensorflow.python.framework import (  # pylint:disable=no-name-in-module
      errors_impl as tf_errors)
  
from lib.image import hex_to_rgb
  from lib.training import PreviewDataGenerator, TrainingDataGenerator
  from lib.training.generator import BatchType, DataGenerator
  from lib.utils import FaceswapError, get_backend, get_folder, get_image_paths, get_tf_version
  from plugins.train._config import Config
  
keep keep add keep keep keep keep keep
 Preview - Allow mask color/opacity configuration
  - typing: lib.config
  - docs: lib.config
  - bugfix: popup_configure - keep assets on top 
plugins/train/trainer/_base.py