code: def apply_conv_norm(
      module: nn.Module, norm_type: str = "identity"
  ) -> nn.Module:
      _verify_norm_type(norm_type)
      if norm_type == "spectral":
          return spectral_norm(module)
      else:
          return module
  
imports: import math
  from functools import partial
  from typing import Callable, List, Optional, Union
  import torch
  import torch.nn
  import torch.nn.functional
  from torch import Tensor
  from torch.nn.utils.parametrizations import spectral_norm
  from ._discriminator import Discriminator
label_desc: Use image node_id 2 for calling a global function with example usage: apply_conv_norm(module, norm_type) and returns: spectral_norm, module