  from keras import Input, Model, backend as K
  
  from numpy.testing import assert_allclose
  
  from lib.model import layers, normalization
  from lib.utils import get_backend
  from tests.utils import has_arg
  
  CONV_SHAPE = (3, 3, 256, 2048)
  CONV_ID = get_backend().upper()
 
</s> remove     if isinstance(layer, (layers.ReflectionPadding2D, normalization.InstanceNormalization)):
        layer.build(input_shape)
 </s> add     layer.build(input_shape)
from lib.model import layers
  from keras import Input, Model, backend as K
  
  from numpy.testing import assert_allclose
  
from lib.model import layers
  from lib.utils import get_backend
  from tests.utils import has_arg
  
  CONV_SHAPE = (3, 3, 256, 2048)
  CONV_ID = get_backend().upper()
keep keep keep keep replace keep keep keep keep keep
 lib.model.normalization:
  - refactor
  - add LayerNorm
  - add RMSNorm 
tests/lib/model/layers_test.py