1d4823c0ec446e93d00df8ca654db4b45b63b3d4,rllib/policy/tests/test_compute_log_likelihoods.py,,do_test_log_likelihood,#Any#Any#Any#Any#Any#Any#,18

Before Change

    prev_r = None if prev_a is None else np.array(0.0)

    // Test against all frameworks.
    for fw in ["tf", "eager", "torch"]:
        if run in [dqn.DQNTrainer, sac.SACTrainer] and fw == "torch":
            continue
        print("Testing {} with framework={}".format(run, fw))
        config["eager"] = fw == "eager"
        config["use_pytorch"] = fw == "torch"

        eager_ctx = None
        if fw == "tf":
            assert not tf.executing_eagerly()
        elif fw == "eager":
            eager_ctx = eager_mode()
            eager_ctx.__enter__()
            assert tf.executing_eagerly()

        trainer = run(config=config, env=env)

        policy = trainer.get_policy()
        vars = policy.get_weights()

After Change

    prev_r = None if prev_a is None else np.array(0.0)

    // Test against all frameworks.
    for fw in framework_iterator(config):
        if run in [dqn.DQNTrainer, sac.SACTrainer] and fw == "torch":
            continue

Instances