def METHOD_NAME(
    logpsi: Callable, params: PyTree, x: Array, data: Optional[PyTree]
):
    result = [
        data.coeffs[i] * op.METHOD_NAME(logpsi, params, x, op_data)
        for i, (op, op_data) in enumerate(zip(data.ops, data.op_data))
    ]
    return sum(result)