def _create_a3plus_config()

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

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

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