def transformed_model_fn(*args, **kwargs):
    (mapped_args, mapped_kwargs, fixed_obs) = obs_to_model_args_fn(*args, **kwargs)
    return condition(model, data=fixed_obs)(*mapped_args, **mapped_kwargs)
def transformed_model_fn(*args, **kwargs):
    (mapped_args, mapped_kwargs, fixed_obs) = obs_to_model_args_fn(*args, **kwargs)
    return condition(model, data=fixed_obs)(*mapped_args, **mapped_kwargs)