c009c178f6b8adaff9bb8e9cca2d0797d3d14425,rllib/rollout.py,,rollout,#Any#Any#Any#Any#Any#Any#Any#,345

Before Change

        state_init = {p: m.get_initial_state() for p, m in policy_map.items()}
        use_lstm = {p: len(s) > 0 for p, s in state_init.items()}
    else:
        env = gym.make(env_name)
        multiagent = False
        try:
            policy_map = {DEFAULT_POLICY_ID: agent.policy}
        except AttributeError:

After Change

        use_lstm = {p: len(s) > 0 for p, s in state_init.items()}
    else:
        from gym import envs
        if envs.registry.env_specs.get(agent.config["env"]):
            // if environment is gym environment, load from gym
            env = gym.make(agent.config["env"])
        else:
            // if environment registered ray environment, load from ray
            env_creator = _global_registry.get(ENV_CREATOR,
                                               agent.config["env"])
            env_context = EnvContext(
                agent.config["env_config"] or {}, worker_index=0)
            env = env_creator(env_context)
        multiagent = False
        try:
            policy_map = {DEFAULT_POLICY_ID: agent.policy}
        except AttributeError:

Instances