def _create_a4_config()

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

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

def _create_a4_config():
  return config_pb2.ASeriesConfig(
      instance_type="a4-highgpu-8g",
      second_pass_yaml_path="a4/nccl_secondpass.yaml",
      nccl_test_command_template=(
          "bash /scripts/run-nccl-combined-plugins.sh rdma {benchmark}"
          " {ld_library_path} 8 eth1,eth2,eth3,eth4,eth5,eth6,eth7,eth8"
          " {start_message_size} {end_message_size} {nhosts} 3"
      ),
      default_threshold=120,
      ld_library_path="/usr/local/gib/lib64:/usr/local/nvidia/lib64/"
  )