def write_run()

in dags/map_reproducibility/utils/benchmarkdb_utils.py [0:0]

• 35 lines of code
• 2 McCabe index (conditional complexity)

def write_run(
    model_id: str,
    hardware_id: str,
    software_id: str,
    number_of_nodes: int,
    number_of_chips: int,
    container_image_name: str,
    global_batch_size: int,
    precision: str,
    optimizer: str,
    seq_length: int,
    median_step_time: float,
    e2e_time: float,
    number_of_steps: int,
    mfu: float,
    tokens_per_second: float,
    writer_path: str,
    run_success: bool = True,  # True because if mfu is none, writing to db will fail anyway.
    run_type: str = "perf_regression",
    run_release_status: str = "local",
    other_metrics_in_json: str = "",
    nccl_driver_nickname: str = None,
    env_variables: str = "",
    framework_config_in_json: str = "",
    xla_flags: str = "",
    topology: str = "",
    dataset: str = "",
    num_of_superblock: int = None,
    update_person_ldap: str = getpass.getuser(),
    comment: str = "",
    is_test: bool = False,
    logs_profile="",
    gcs_metrics_bucket="",
    workload_others="",
    experiment_id="",