def METHOD_NAME(self, args, no_encoder_attn=False, layer_idx=None):
    return HeadSelectionTransformerDecoderLayer(
        args,
        layer_idx,
        self.self_attn_head_selector,
        self.enc_attn_head_selector,
        no_encoder_attn=no_encoder_attn
    )