def METHOD_NAME(log, model):
    model.fit(log)
    assert len(list(model.model.parameters())) == 12
    param_shapes = [
        (3, 2),
        (4, 2),
        (4, 1),
        (3, 1),
        (3, 2),
        (4, 2),
        (4, 1),
        (3, 1),
        (2, 4),
        (2,),
        (1, 4),
        (1,),
    ]
    for i, parameter in enumerate(model.model.parameters()):
        assert param_shapes[i] == tuple(parameter.shape)