def METHOD_NAME(self):
    class DummyModel(nn.Module):
        @resolve_param("activation_in_head", ActivationsTypeFactory())
        def __init__(self, activation_in_head):
            super().__init__()
            self.activation_in_head = activation_in_head()
    model = DummyModel(activation_in_head=nn.LeakyReLU)
    self.assertIsInstance(model.activation_in_head, nn.LeakyReLU)