1d4823c0ec446e93d00df8ca654db4b45b63b3d4,rllib/tests/test_model_imports.py,,model_import_test,#Any#Any#Any#,127

Before Change

    agent_cls = get_agent_class(algo)

    for fw in ["tf", "torch"]:
        print("framework={}".format(fw))

        config["use_pytorch"] = fw == "torch"
        config["eager"] = fw == "eager"
        config["model"]["custom_model"] = "keras_model" if fw != "torch" else \
            "torch_model"

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

        agent = agent_cls(config, env)

        def current_weight(agent):
            if fw == "tf":

After Change

    agent_cls = get_agent_class(algo)

    for fw in framework_iterator(config, ["tf", "torch"]):
        config["model"]["custom_model"] = "keras_model" if fw != "torch" else \
            "torch_model"

Instances