  #!/usr/bin/env python3
  """ Conditional imports depending on whether the AMD version is installed or not """
  
  from lib.utils import get_backend
  from .normalization_common import *
  
  if get_backend() == "amd":
      from .normalization_plaid import *
  else:
      from .normalization_tf import *
 
</s> remove     from .normalization_plaid import *
 </s> add     from .normalization_plaid import LayerNormalization, RMSNormalization </s> remove     from .normalization_tf import *
 </s> add     from .normalization_tf import LayerNormalization, RMSNormalization </s> remove from keras import regularizers
 </s> add import numpy as np
from .normalization_common import AdaInstanceNormalization
from .normalization_common import GroupNormalization
from .normalization_common import InstanceNormalization

  #!/usr/bin/env python3
  """ Conditional imports depending on whether the AMD version is installed or not """
  
  from lib.utils import get_backend
from .normalization_common import AdaInstanceNormalization
from .normalization_common import GroupNormalization
from .normalization_common import InstanceNormalization

  
  if get_backend() == "amd":
      from .normalization_plaid import *
  else:
      from .normalization_tf import *
keep keep keep keep replace keep keep keep keep keep
 Unit tests - lib.model.normalization
  - __init__.py - Explicit imports (to fix pytests)
  - Add tests for all existing normalization layers
  - fix GroupNormalization for tensorflow backend 
lib/model/normalization/__init__.py