          self._use_depthwise = use_depthwise
  
          self._assert_arguments()
  
      def _assert_arguments(self):
          """ Validate the given arguments. """
          assert self._normalization in ("batch", "instance", None), (
              "normalization should be 'batch', 'instance' or None")
          assert self._activation in ("leakyrelu", "swish", "prelu", None), (
              "activation should be 'leakyrelu', 'prelu', 'swish' or None")
 
</s> remove     check_icnr_init: `bool`, optional
        ``True`` if the user configuration options should be checked to apply ICNR initialization
        to the layer. This should only be passed in from :class:`UpscaleBlock` layers.
        Default: ``False``
 </s> add     is_upscale: `bool`, optional
        ``True`` if the convolution is being called from an upscale layer. This causes the instance
        to check the user configuration options to see if ICNR initialization has been selected and
        should be applied. This should only be passed in as ``True`` from :class:`UpscaleBlock`
        layers. Default: ``False`` </s> remove     def __call__(self, inputs):
 </s> add     def __call__(self, inputs: Tensor) -> Tensor: </s> remove def _get_name(name):
 </s> add def _get_name(name: str) -> str: </s> remove     def __call__(self, inputs):
 </s> add     def __call__(self, inputs: Tensor) -> Tensor: </s> remove     def __init__(self, filters, kernel_size=3, padding="same", **kwargs):
 </s> add     def __init__(self,
                 filters: int,
                 kernel_size: Union[int, Tuple[int]] = 3,
                 padding: str = "same",
                 **kwargs) -> None:
    def _assert_arguments(self) -> None:
          self._use_depthwise = use_depthwise
  
          self._assert_arguments()
  
    def _assert_arguments(self) -> None:
          """ Validate the given arguments. """
          assert self._normalization in ("batch", "instance", None), (
              "normalization should be 'batch', 'instance' or None")
          assert self._activation in ("leakyrelu", "swish", "prelu", None), (
              "activation should be 'leakyrelu', 'prelu', 'swish' or None")
keep keep keep keep replace keep keep keep keep keep
 lib.model.nnblocks - typing + cleanup upscales 
lib/model/nn_blocks.py