8e6fc06cc156510d37decf430cf3df0853285779,softlearning/value_functions/utils.py,,get_Q_function_from_variant,#Any#Any#,4

Before Change

def get_Q_function_from_variant(variant, env):
    layer_size = variant["value_fn_params"]["layer_size"]

    Qs = tuple(
        NNQFunction(
            observation_shape=env.observation_space.shape,
            action_shape=env.action_space.shape,
            hidden_layer_sizes=(layer_size, layer_size),
            name="qf{}".format(i))
        for i in range(2))

    return Qs

After Change

def get_Q_function_from_variant(variant, env):
    Q_params = variant["Q_params"]
    Q_function = Q_FUNCTION_FUNCTIONS[Q_params["type"]](variant, env)
    return Q_function

Instances