def _create_a3_config()

in src/nccl_healthcheck/config.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def _create_a3_config():
  return config_pb2.ASeriesConfig(
      instance_type="a3-highgpu-8g",
      second_pass_yaml_path="a3/nccl_secondpass.yaml",
      nccl_test_command_template=(
          "bash /scripts/run-nccl-combined-plugins.sh gpudirect {benchmark}"
          " {ld_library_path} 8 eth1,eth2,eth3,eth4"
          " {start_message_size} {end_message_size} {nhosts} 1 {iterations}"
      ),
      default_threshold=60,
      ld_library_path="/usr/local/tcpx/lib64:/usr/local/nvidia/lib64/",
  )