  from lib.utils import get_backend
  from tests.utils import has_arg
  
  CONV_SHAPE = (3, 3, 256, 2048)
  CONV_ID = get_backend().upper()
  
  
  def layer_test(layer_cls, kwargs={}, input_shape=None, input_dtype=None,
                 input_data=None, expected_output=None,
 
</s> remove import keras
from keras import backend as K

 </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.utils import get_backend
  from tests.utils import has_arg
  
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


  CONV_SHAPE = (3, 3, 256, 2048)
  CONV_ID = get_backend().upper()
  
  
  def layer_test(layer_cls, kwargs={}, input_shape=None, input_dtype=None,
                 input_data=None, expected_output=None,
keep keep add keep 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/layers_test.py