1d4823c0ec446e93d00df8ca654db4b45b63b3d4,rllib/agents/sac/tests/test_sac.py,TestSAC,test_sac_compilation,#TestSAC#,11

Before Change

        // eager (discrete and cont. actions).
        for fw in ["eager", "tf", "torch"]:
            print("framework={}".format(fw))
            if fw == "torch":
                continue
            config["eager"] = fw == "eager"
            config["use_pytorch"] = fw == "torch"
            for env in [
                    "CartPole-v0",
                    "Pendulum-v0",
            ]:

After Change

    def test_sac_compilation(self):
        Test whether an SACTrainer can be built with all frameworks.
        ray.init()
        config = sac.DEFAULT_CONFIG.copy()
        config["num_workers"] = 0  // Run locally.
        num_iterations = 1

        // eager (discrete and cont. actions).
        for _ in framework_iterator(config, ["tf", "eager"]):
            for env in [
                    "CartPole-v0",
                    "Pendulum-v0",

Instances