  from importlib import import_module
  
  from lib.gpu_stats import set_exclude_devices, GPUStats
  from lib.logger import crash_log, log_setup
  from lib.utils import (FaceswapError, get_backend, get_tf_version, KerasFinder, safe_shutdown,
                         set_backend, set_system_verbosity)
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  
  
 
</s> remove from keras import backend as K
from keras import initializers
 </s> add  </s> remove from lib.utils import get_backend, get_keras_custom_objects as get_custom_objects
 </s> add from lib.utils import get_backend

if get_backend() == "amd":
    from keras.utils import get_custom_objects  # pylint:disable=no-name-in-module
    from keras import backend as K
    from keras import initializers
else:
    # Ignore linting errors from Tensorflow's thoroughly broken import system
    from tensorflow.keras.utils import get_custom_objects  # noqa pylint:disable=no-name-in-module,import-error
    from tensorflow.keras import initializers, backend as K  # noqa pylint:disable=no-name-in-module,import-error </s> remove from keras.initializers import RandomNormal
from keras.layers import Input, LeakyReLU

 </s> add 
from lib.utils import (FaceswapError, get_backend, get_tf_version, safe_shutdown,
  from importlib import import_module
  
  from lib.gpu_stats import set_exclude_devices, GPUStats
  from lib.logger import crash_log, log_setup
from lib.utils import (FaceswapError, get_backend, get_tf_version, safe_shutdown,
                         set_backend, set_system_verbosity)
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  
  
keep keep keep keep replace keep keep keep keep keep
 Update all Keras Imports to be conditional (#1214)

* Remove custom keras importer

* first round keras imports fix

* launcher.py: Remove KerasFinder references

* 2nd round keras imports update (lib and extract)

* 3rd round keras imports update (train)

* remove KerasFinder from tests

* 4th round keras imports update (tests) 
lib/cli/launcher.py