  from lib.model import losses
  from lib.utils import get_backend
  
  
  _PARAMS = [(losses.GeneralizedLoss(), (2, 16, 16)),
             (losses.GradientLoss(), (2, 16, 16)),
             # TODO Make sure these output dimensions are correct
             (losses.GMSDLoss(), (2, 1, 1)),
 
</s> remove _IDS = ["{}[{}]".format(loss, get_backend().upper()) for loss in _IDS]
 </s> add _IDS = [f"{loss}[{get_backend().upper()}]" for loss in _IDS] </s> remove from numpy.testing import assert_allclose

from keras import backend as K
from keras import losses as k_losses
from keras.layers import Conv2D
from keras.models import Sequential
from keras.optimizers import Adam
 </s> add  </s> remove from keras import losses as k_losses
from keras import backend as K
from keras.layers import Input
from keras.models import load_model, Model as KModel

 </s> add  </s> remove from keras import regularizers, models, layers

 </s> add 
if get_backend() == "amd":
    from keras import backend as K, losses as k_losses
else:
    # Ignore linting errors from Tensorflow's thoroughly broken import system
    from tensorflow.keras import backend as K, losses as k_losses  # pylint:disable=import-error
  from lib.model import losses
  from lib.utils import get_backend
  
if get_backend() == "amd":
    from keras import backend as K, losses as k_losses
else:
    # Ignore linting errors from Tensorflow's thoroughly broken import system
    from tensorflow.keras import backend as K, losses as k_losses  # pylint:disable=import-error
  
  _PARAMS = [(losses.GeneralizedLoss(), (2, 16, 16)),
             (losses.GradientLoss(), (2, 16, 16)),
             # TODO Make sure these output dimensions are correct
             (losses.GMSDLoss(), (2, 1, 1)),
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/losses_test.py