1d4823c0ec446e93d00df8ca654db4b45b63b3d4,rllib/utils/schedules/tests/test_schedules.py,TestSchedules,test_piecewise_schedule,#TestSchedules#,94

Before Change

            endpoints=[(0, 50.0), (25, 100.0), (30, 200.0)],
            outside_value=14.5)
        for fw in ["tf", "torch", None]:
            config["framework"] = fw
            piecewise = from_config(PiecewiseSchedule, config)
            for t, e in zip(ts, expected):
                out = piecewise(t)
                check(out, e, decimals=4)

        // Test eager as well.
        with eager_mode():
            config["framework"] = "tf"
            piecewise = from_config(PiecewiseSchedule, config)
            for t, e in zip(ts, expected):
                out = piecewise(t)
                check(out, e, decimals=4)


if __name__ == "__main__":
    import pytest
    import sys
    sys.exit(pytest.main(["-v", __file__]))

After Change

            endpoints=[(0, 50.0), (25, 100.0), (30, 200.0)],
            outside_value=14.5)

        for fw in framework_iterator(
                frameworks=["tf", "eager", "torch", None]):
            fw_ = fw if fw != "eager" else "tf"
            piecewise = from_config(PiecewiseSchedule, config, framework=fw_)
            for t, e in zip(ts, expected):
                out = piecewise(t)
                check(out, e, decimals=4)

Instances