  from lib.model import nn_blocks
  from lib.utils import get_backend
  
  
  def block_test(layer_func, kwargs={}, input_shape=None):
      """Test routine for faceswap neural network blocks.
  
      Tests are simple and are to ensure that the blocks compile on both tensorflow
 
</s> remove from keras import Input, Model, backend as K
 </s> add  </s> remove def get_keras_custom_objects():
    """ Wrapper to obtain keras.utils.get_custom_objects from correct location depending on
    backend used and tensorflow version. """
    # pylint:disable=no-name-in-module,import-outside-toplevel
    if get_backend() == "amd" or get_tf_version() < 2.8:
        from keras.utils import get_custom_objects
    else:
        from keras.utils.generic_utils import get_custom_objects
    return get_custom_objects()


 </s> add 
if get_backend() == "amd":
    from keras import Input, Model, backend as K
else:
    # Ignore linting errors from Tensorflow's thoroughly broken import system
    from tensorflow.keras import Input, Model, backend as K  # pylint:disable=import-error

  from lib.model import nn_blocks
  from lib.utils import get_backend
  
if get_backend() == "amd":
    from keras import Input, Model, backend as K
else:
    # Ignore linting errors from Tensorflow's thoroughly broken import system
    from tensorflow.keras import Input, Model, backend as K  # pylint:disable=import-error

  
  def block_test(layer_func, kwargs={}, input_shape=None):
      """Test routine for faceswap neural network blocks.
  
      Tests are simple and are to ensure that the blocks compile on both tensorflow
keep keep add 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) 
tests/lib/model/nn_blocks_test.py