  Adapted from Keras tests.
  """
  
  from keras import regularizers
  import pytest
  
  from lib.model import normalization
 
</s> remove from lib.model import layers, normalization
 </s> add from lib.model import layers </s> remove     if isinstance(layer, (layers.ReflectionPadding2D, normalization.InstanceNormalization)):
        layer.build(input_shape)
 </s> add     layer.build(input_shape)
from itertools import product
  Adapted from Keras tests.
  """
from itertools import product
  
  from keras import regularizers
  import pytest
  
  from lib.model import normalization
keep add keep keep keep keep keep
 lib.model.normalization:
  - refactor
  - add LayerNorm
  - add RMSNorm 
tests/lib/model/normalization_test.py