hunk.code_window:     @override(BaseEnv)
       def try_reset(self,
                     env_id: Optional[EnvID] = None) -> Optional[MultiAgentDict]:
           obs = self.env_states[env_id].reset()
           assert isinstance(obs, dict), "Not a multi-agent obs"
           if obs is not None and env_id in self.dones:
               self.dones.remove(env_id)
  
hunk.labels: keep keep keep replace keep keep keep keep
hunk.after_edit:                   env_id: Optional[EnvID] = None) -> Optional[MultiEnvDict]:
  
hunk.file_path: rllib/env/multi_agent_env.py
hunk.type: replace
file: cluster_name: default
  min_workers: 0
  max_workers: 0
  docker:
      image: ""
      container_name: ""
  upscaling_speed: 1.0
  idle_timeout_minutes: 5
  provider:
      type: local
      head_ip: YOUR_HEAD_NODE_HOSTNAME
      worker_ips: []
  auth:
      ssh_user: YOUR_USERNAME
      ssh_private_key: ~/.ssh/id_rsa
  head_node: {}
  worker_nodes: {}
  file_mounts:
       "/tmp/ray_sha": "/YOUR/LOCAL/RAY/REPO/.git/refs/heads/YOUR_BRANCH"
  setup_commands: []
  head_setup_commands: []
  worker_setup_commands: []
  setup_commands:
      - source activate ray && test -e ray || git clone https://github.com/YOUR_GITHUB/ray.git
      - source activate ray && cd ray && git fetch && git reset --hard `cat /tmp/ray_sha`
  #    - source activate ray && cd ray/python && pip install -e .
  head_start_ray_commands:
      - source activate ray && ray stop
      - source activate ray && ulimit -c unlimited && ray start --head --port=6379 --autoscaling-config=~/ray_bootstrap_config.yaml
  worker_start_ray_commands:
      - source activate ray && ray stop
      - source activate ray && ray start --address=$RAY_HEAD_IP:6379
  