hunk.code_window: @pytest.mark.parametrize(
       "ray_start_cluster_head", [
           generate_system_config_map(
               num_heartbeats_timeout=20, ping_gcs_rpc_server_max_retries=60)
       ],
       indirect=True)
   def test_node_failure_detector_when_gcs_server_restart(ray_start_cluster_head):
       """Checks that the node failure detector is correct when gcs server restart.
  
hunk.labels: keep keep keep keep keep replace keep keep keep keep
hunk.after_edit:             num_heartbeats_timeout=2, ping_gcs_rpc_server_max_retries=60)
  
hunk.file_path: python/ray/tests/test_gcs_fault_tolerance.py
hunk.type: replace
file: # An unique identifier for the head node and workers of this cluster.
  cluster_name: sgd-coco-pytorch
  
  # The maximum number of workers nodes to launch in addition to the head
  # node. This takes precedence over min_workers. min_workers default to 0.
  min_workers: 1
  max_workers: 1
  
  # Cloud-provider specific configuration.
  provider:
      type: aws
      region: us-east-1
      availability_zone: us-east-1c
  
  # How Ray will authenticate with newly launched nodes.
  auth:
      ssh_user: ubuntu
  
  
  head_node:
      InstanceType: p3.8xlarge
      ImageId: ami-0698bcaf8bd9ef56d
      InstanceMarketOptions:
          MarketType: spot
      BlockDeviceMappings:
          - DeviceName: /dev/sda1
            Ebs:
                VolumeSize: 300
  
  
  worker_nodes:
      InstanceType: p3.8xlarge
      ImageId: ami-0698bcaf8bd9ef56d
      InstanceMarketOptions:
          MarketType: spot
      BlockDeviceMappings:
          - DeviceName: /dev/sda1
            Ebs:
                VolumeSize: 300
  
  setup_commands:
      # This replaces the standard anaconda Ray installation
      - ray || pip install -U https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-2.0.0.dev0-cp36-cp36m-manylinux2014_x86_64.whl
      # Uncomment this and the filemount to update the Ray installation with your local Ray code
      # - rm -rf ./anaconda3/lib/python3.6/site-packages/ray/util/sgd/
      # - cp -rf ~/sgd ./anaconda3/lib/python3.6/site-packages/ray/util/
      - pip install -q tqdm
  
      # Installing this without -U to make sure we don't replace the existing Ray installation
      - pip install ray[rllib]
      - pip install -U ipdb torch torchvision pycocotools
      # Install Apex if needed.
      - git clone https://github.com/NVIDIA/apex; cd apex && pip install -v --no-cache-dir  ./ || true
  
  
  file_mounts: {
      # # This should point to ray/python/ray/util/sgd.
      # ~/sgd: ../../../../sgd,
  }
  