83e06cd30a45245c2cb0e9f4bd924224b1581554,rllib/policy/eager_tf_policy.py,,_convert_to_numpy,#Any#,36

Before Change

    if x is None:
        return None
    try:
        return x.numpy()
    except AttributeError:
        raise TypeError(
            ("Object of type {} has no method to convert to numpy.").format(
                type(x)))

After Change

    if x is None:
        return None
    try:
        return tf.nest.map_structure(lambda component: component.numpy(), x)
    except AttributeError:
        raise TypeError(
            ("Object of type {} has no method to convert to numpy.").format(
                type(x)))

Instances