1d4823c0ec446e93d00df8ca654db4b45b63b3d4,rllib/utils/schedules/tests/test_schedules.py,TestSchedules,test_polynomial_schedule,#TestSchedules#,52

Before Change

            power=2.0)
        for fw in ["tf", "torch", None]:
            config["framework"] = fw
            polynomial = from_config(config)
            for t in ts:
                out = polynomial(t)
                check(out, 0.5 + (2.0 - 0.5) * (1.0 - t / 100)**2, decimals=4)

        // Test eager as well.
        with eager_mode():
            config["framework"] = "tf"
            polynomial = from_config(config)
            for t in ts:

After Change

            final_p=0.5,
            power=2.0)

        for fw in framework_iterator(
                frameworks=["tf", "eager", "torch", None]):
            fw_ = fw if fw != "eager" else "tf"
            polynomial = from_config(config, framework=fw_)
            for t in ts:
                out = polynomial(t)
                check(out, 0.5 + (2.0 - 0.5) * (1.0 - t / 100)**2, decimals=4)

Instances