def __init__(self, filters: int, pool_size: int):
        """Initializes PANNWavBlock.

        Args:
          filters: the number of filters.
          pool_size: kernel size and stride of the 1D max-pooling.
        """
        super().__init__()

        for dilation_rate in [1, 2]:
            self.add(tf.keras.layers.Conv1D(
                filters=filters,
                kernel_size=3,
                strides=1,
                dilation_rate=dilation_rate,
                padding='SAME',
                use_bias=False, ))
            self.add(tf.keras.layers.BatchNormalization())
            self.add(tf.keras.layers.ReLU())

        self.add(tf.keras.layers.MaxPool1D(
            pool_size=pool_size, strides=pool_size))
Initializes PANNWavBlock.

        Args:
          filters: the number of filters.
          pool_size: kernel size and stride of the 1D max-pooling.
        
leaf_audio/models.py