  
  This model is heavily documented as it acts as a template that other model plugins can be developed
  from.
  """
  from keras.layers import Dense, Flatten, Reshape, Input
  
  from lib.model.nn_blocks import Conv2DOutput, Conv2DBlock, UpscaleBlock
  from ._base import KerasModel, ModelBase
  
  
 
</s> remove from keras.layers import Concatenate, Dense, Flatten, Input, LeakyReLU, Reshape

 </s> add 
  
  This model is heavily documented as it acts as a template that other model plugins can be developed
  from.
  """

  
  from lib.model.nn_blocks import Conv2DOutput, Conv2DBlock, UpscaleBlock
  from ._base import KerasModel, ModelBase
  
  
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) 
plugins/train/model/original.py