id: 1
unit: def convert_qwen_state_dict_from_megatron_to_vllm()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 211 LOC
McCabe index: 34
number of parameters: 3

id: 2
unit: def convert_checkpoint_from_megatron_to_transformers()
file: chatlearn/tools/megatron_to_hf.py
start line: 0
end line: 0
size: 146 LOC
McCabe index: 30
number of parameters: 1

id: 3
unit: def _init_workers_ray()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/ray_gpu_executor.py
start line: 0
end line: 0
size: 124 LOC
McCabe index: 47
number of parameters: 3

id: 4
unit: def _validate_params()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 118 LOC
McCabe index: 44
number of parameters: 1

id: 5
unit: def convert_llama_state_dict_from_megatron_to_vllm()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 115 LOC
McCabe index: 18
number of parameters: 3

id: 6
unit: def build_rank_mapping_two_stage()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 113 LOC
McCabe index: 37
number of parameters: 2

id: 7
unit: def _init_workers_ray()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/ray_gpu_executor.py
start line: 0
end line: 0
size: 112 LOC
McCabe index: 43
number of parameters: 3

id: 8
unit: def learn()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 99 LOC
McCabe index: 18
number of parameters: 1

id: 9
unit: def convert_llama_state_dict_from_mcore_to_vllm()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 96 LOC
McCabe index: 14
number of parameters: 3

id: 10
unit: def _init_args()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 95 LOC
McCabe index: 8
number of parameters: 1

id: 11
unit: def preprocess_compute()
file: chatlearn/runtime/decorator.py
start line: 0
end line: 0
size: 94 LOC
McCabe index: 43
number of parameters: 2

id: 12
unit: def broadcast_parameter_two_stage()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 91 LOC
McCabe index: 26
number of parameters: 8

id: 13
unit: def __init__()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 90 LOC
McCabe index: 9
number of parameters: 4

id: 14
unit: def _add_request_internal()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 89 LOC
McCabe index: 19
number of parameters: 4

id: 15
unit: def __init__()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 88 LOC
McCabe index: 5
number of parameters: 4

id: 16
unit: def forward()
file: chatlearn/models/megatron/ops/policy_gradient.py
start line: 0
end line: 0
size: 87 LOC
McCabe index: 15
number of parameters: 8

id: 17
unit: def _synchronize_all_moe_parameters()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 87 LOC
McCabe index: 8
number of parameters: 4

id: 18
unit: def load_weights()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 80 LOC
McCabe index: 29
number of parameters: 3

id: 19
unit: def regroup_qkv_tp_slices()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 80 LOC
McCabe index: 18
number of parameters: 4

id: 20
unit: def build_scheduler()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 78 LOC
McCabe index: 18
number of parameters: 1

id: 21
unit: def check_log_files_and_publish_updates()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 76 LOC
McCabe index: 19
number of parameters: 1

id: 22
unit: def _parse_params()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 75 LOC
McCabe index: 28
number of parameters: 2

id: 23
unit: def log_metrics_stats()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 75 LOC
McCabe index: 16
number of parameters: 2

id: 24
unit: def init()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/llm.py
start line: 0
end line: 0
size: 72 LOC
McCabe index: 4
number of parameters: 30

id: 25
unit: def convert_layer_to_lora()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 72 LOC
McCabe index: 39
number of parameters: 7

id: 26
unit: def validate_sync_results()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 71 LOC
McCabe index: 18
number of parameters: 6

id: 27
unit: def build_rank_mapping_for_ep()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 70 LOC
McCabe index: 28
number of parameters: 2

id: 28
unit: def onload_weights()
file: chatlearn/models/megatron/memory_manager/trainer_v3.py
start line: 0
end line: 0
size: 69 LOC
McCabe index: 19
number of parameters: 1

id: 29
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 62 LOC
McCabe index: 3
number of parameters: 13

id: 30
unit: def onload_weights()
file: chatlearn/models/megatron/memory_manager/trainer_v1v2.py
start line: 0
end line: 0
size: 62 LOC
McCabe index: 17
number of parameters: 1

id: 31
unit: def sync_broadcast_multi_threads()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 59 LOC
McCabe index: 19
number of parameters: 10

id: 32
unit: def alltoall_routed_experts_from_hep()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 58 LOC
McCabe index: 10
number of parameters: 4

id: 33
unit: def place_gpu_models()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 58 LOC
McCabe index: 31
number of parameters: 3

id: 34
unit: def map_src_to_dst()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 25
number of parameters: 1

id: 35
unit: def build_dataloader()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 7
number of parameters: 10

id: 36
unit: def onload_weights()
file: chatlearn/models/megatron/memory_manager/trainer_v4.py
start line: 0
end line: 0
size: 55 LOC
McCabe index: 13
number of parameters: 1

id: 37
unit: def generate_vllm()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 54 LOC
McCabe index: 18
number of parameters: 4

id: 38
unit: def build_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v1v2.py
start line: 0
end line: 0
size: 54 LOC
McCabe index: 14
number of parameters: 1

id: 39
unit: def add_checkpointing_args()
file: chatlearn/tools/megatron_to_hf.py
start line: 0
end line: 0
size: 54 LOC
McCabe index: 1
number of parameters: 1

id: 40
unit: def regroup_params_to_sync()
file: chatlearn/synchronizer/base.py
start line: 0
end line: 0
size: 54 LOC
McCabe index: 19
number of parameters: 5

id: 41
unit: def __init__()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 8
number of parameters: 6

id: 42
unit: def update_log_filenames()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 16
number of parameters: 1

id: 43
unit: def _load_base_checkpoint()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 14
number of parameters: 2

id: 44
unit: def process_model_outputs()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 51 LOC
McCabe index: 21
number of parameters: 3

id: 45
unit: def allgather_routed_experts_from_hep()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 50 LOC
McCabe index: 9
number of parameters: 5

id: 46
unit: def execute_step()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 11
number of parameters: 2

id: 47
unit: def _set_sync_param_names()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 17
number of parameters: 7

id: 48
unit: def logging_summary()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 46 LOC
McCabe index: 25
number of parameters: 2

id: 49
unit: def _synchronize_params_except_routed_experts()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 45 LOC
McCabe index: 3
number of parameters: 4

id: 50
unit: def __iter__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 44 LOC
McCabe index: 14
number of parameters: 1

id: 51
unit: def model_setup()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 44 LOC
McCabe index: 3
number of parameters: 1

id: 52
unit: def _init_distributed_environment()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 10
number of parameters: 1

id: 53
unit: def setup_vllm()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 9
number of parameters: 2

id: 54
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 8
number of parameters: 4

id: 55
unit: def forward()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 5
number of parameters: 2

id: 56
unit: def build_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v4.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 11
number of parameters: 1

id: 57
unit: def setup_rank_mapping()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 12
number of parameters: 1

id: 58
unit: def remote()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 13
number of parameters: 1

id: 59
unit: def sync_broadcast_two_stage()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 12
number of parameters: 7

id: 60
unit: def model_setup()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 6
number of parameters: 1

id: 61
unit: def generate_step_one_model_internal()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 12
number of parameters: 11

id: 62
unit: def open_closed_files()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 11
number of parameters: 1

id: 63
unit: def load_checkpoint()
file: chatlearn/utils/megatron_utils.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 11
number of parameters: 2

id: 64
unit: def set_vllm_pp_layer_partition()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 11
number of parameters: 1

id: 65
unit: def load_weights()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 14
number of parameters: 3

id: 66
unit: def __init__()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 7
number of parameters: 7

id: 67
unit: def forward()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 6
number of parameters: 2

id: 68
unit: def set_vllm_pp_layer_partition()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 11
number of parameters: 1

id: 69
unit: def __init__()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 5
number of parameters: 7

id: 70
unit: def compute_loop_one_model()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 18
number of parameters: 3

id: 71
unit: def get_src_and_dst_dp_ranks()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 17
number of parameters: 2

id: 72
unit: def init_engine_args()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 7
number of parameters: 1

id: 73
unit: def rebatch_all_merged_data()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 9
number of parameters: 4

id: 74
unit: def execute()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 12
number of parameters: 2

id: 75
unit: def _get_sampling_params()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 12
number of parameters: 2

id: 76
unit: def fix_qkv_ordering()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 9
number of parameters: 2

id: 77
unit: def __init__()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 4
number of parameters: 7

id: 78
unit: def sync()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 3
number of parameters: 4

id: 79
unit: def _synchronize_routed_experts()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 2
number of parameters: 4

id: 80
unit: def save_checkpoint()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 8
number of parameters: 2

id: 81
unit: def offload_weights()
file: chatlearn/models/megatron/memory_manager/trainer_v3.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 11
number of parameters: 1

id: 82
unit: def get_merged_data()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 14
number of parameters: 6

id: 83
unit: def execute()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 11
number of parameters: 2

id: 84
unit: def start_exit_listener()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 12
number of parameters: 1

id: 85
unit: def _close_all_files()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 12
number of parameters: 1

id: 86
unit: def map_src_to_dst()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 8
number of parameters: 1

id: 87
unit: def build_parameter_group()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 4
number of parameters: 1

id: 88
unit: def __init__()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 3
number of parameters: 3

id: 89
unit: def __init__()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 4
number of parameters: 7

id: 90
unit: def repair_code()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 8
number of parameters: 3

id: 91
unit: def get_next()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 6
number of parameters: 2

id: 92
unit: def copy_to_gpu_buffer()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 7
number of parameters: 3

id: 93
unit: def setup()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 8
number of parameters: 1

id: 94
unit: def sync()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 5
number of parameters: 4

id: 95
unit: def find_model_packing_strategy()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 13
number of parameters: 3

id: 96
unit: def __init__()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 4
number of parameters: 3

id: 97
unit: def bucket_tensors_two_stage_generator()
file: chatlearn/utils/dist_utils.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 11
number of parameters: 4

id: 98
unit: def regroup_by_concat_along_batch()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 12
number of parameters: 1

id: 99
unit: def offload()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 16
number of parameters: 5

id: 100
unit: def __init__()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 4
number of parameters: 1

id: 101
unit: def log()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 11
number of parameters: 7

id: 102
unit: def map_src_to_dst()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 10
number of parameters: 1

id: 103
unit: def __init__()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 4
number of parameters: 6

id: 104
unit: def onload()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 13
number of parameters: 5

id: 105
unit: def get_fsdp_wrap_policy()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 12
number of parameters: 2

id: 106
unit: def load_model()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/loader.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 7
number of parameters: 8

id: 107
unit: def allreduce_word_embedding_grads()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 13
number of parameters: 2

id: 108
unit: def generate_step_one_model()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 17
number of parameters: 12

id: 109
unit: def build_rank_mapping()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 11
number of parameters: 2

id: 110
unit: def load_rank0_state_dict()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 7
number of parameters: 1

id: 111
unit: def preprocess_inputs()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 4
number of parameters: 3

id: 112
unit: def _build_dataloader()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 6
number of parameters: 4

id: 113
unit: def offload_weights()
file: chatlearn/models/megatron/memory_manager/trainer_v1v2.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 6
number of parameters: 1

id: 114
unit: def offload_weights()
file: chatlearn/models/megatron/memory_manager/trainer_v4.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 6
number of parameters: 1

id: 115
unit: def _setup_wandb()
file: chatlearn/schedule/metric_manager.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 5
number of parameters: 1

id: 116
unit: def __init__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 4
number of parameters: 8

id: 117
unit: def wait()
file: chatlearn/utils/future.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 12
number of parameters: 3

id: 118
unit: def model_setup()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 7
number of parameters: 1

id: 119
unit: def model_setup()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 8
number of parameters: 1

id: 120
unit: def schedule()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 7
number of parameters: 1

id: 121
unit: def extract_prompt_components()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/input_preprocess.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 1
number of parameters: 4

id: 122
unit: def eval()
file: chatlearn/runtime/evaluator.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 9
number of parameters: 3

id: 123
unit: def trace()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 16
number of parameters: 3

id: 124
unit: def split_sync_groups()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 9
number of parameters: 3

id: 125
unit: def iter_fixed()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 12
number of parameters: 1

id: 126
unit: def set_dataset()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 6
number of parameters: 5

id: 127
unit: def coalesced_comm_dense_two_stage()
file: chatlearn/utils/dist_utils.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 5
number of parameters: 7

id: 128
unit: def save_zero_three_model()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 9
number of parameters: 4

id: 129
unit: def reset_block_manager()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 9
number of parameters: 1

id: 130
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 3
number of parameters: 4

id: 131
unit: def compute_loop()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 10
number of parameters: 3

id: 132
unit: def place_cpu_models()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 10
number of parameters: 2

id: 133
unit: def _destroy_collective_group()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 7
number of parameters: 2

id: 134
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 1
number of parameters: 4

id: 135
unit: def align_out_queues()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 7
number of parameters: 2

id: 136
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 7
number of parameters: 5

id: 137
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 1
number of parameters: 6

id: 138
unit: def topological_sort()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 13
number of parameters: 1

id: 139
unit: def collective_task_scheduler()
file: chatlearn/synchronizer/scheduler.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 12
number of parameters: 1

id: 140
unit: def add_recv_actor_for_routed_experts()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 1
number of parameters: 3

id: 141
unit: def unflatten_dense_tensors()
file: chatlearn/utils/dist_utils.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 4
number of parameters: 4

id: 142
unit: def init()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/llm.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 1
number of parameters: 0

id: 143
unit: def load_model()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/loader.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 7
number of parameters: 2

id: 144
unit: def free_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v1v2.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 8
number of parameters: 1

id: 145
unit: def backward()
file: chatlearn/models/megatron/ops/policy_gradient.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 1
number of parameters: 2

id: 146
unit: def __init__()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 2
number of parameters: 7

id: 147
unit: def __init__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 3
number of parameters: 8

id: 148
unit: def parse_value()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 9
number of parameters: 1

id: 149
unit: def __init__()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 8
number of parameters: 3

id: 150
unit: def get_parameter_to_sync()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 10
number of parameters: 5

id: 151
unit: def forward()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 4
number of parameters: 2

id: 152
unit: def train()
file: chatlearn/runtime/trainer.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 7
number of parameters: 2

id: 153
unit: def fake_compute()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 6
number of parameters: 2

id: 154
unit: def _validate()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 7
number of parameters: 2

id: 155
unit: def get_synchronizer()
file: chatlearn/synchronizer/__init__.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 8
number of parameters: 2

id: 156
unit: def create_broadcast_group()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 6
number of parameters: 5

id: 157
unit: def sync_broadcast_second_stage_internal()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 7

id: 158
unit: def create_placement_group()
file: chatlearn/schedule/resource_manager.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 8
number of parameters: 4

id: 159
unit: def bucket_tensors()
file: chatlearn/utils/dist_utils.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 8
number of parameters: 2

id: 160
unit: def setup_dataset()
file: chatlearn/runtime/evaluator.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 8
number of parameters: 1

id: 161
unit: def num_iteration()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 2

id: 162
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 7

id: 163
unit: def chatlearn_compare()
file: chatlearn/tools/check_parameter_sync.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 5
number of parameters: 2

id: 164
unit: def add_recv_actor()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 3

id: 165
unit: def sync_broadcast_second_stage()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 6
number of parameters: 7

id: 166
unit: def batching()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 10
number of parameters: 3

id: 167
unit: def offload_weights()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 6
number of parameters: 2

id: 168
unit: def load_checkpoint()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 5
number of parameters: 2

id: 169
unit: def __init__()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 6
number of parameters: 3

id: 170
unit: def from_engine_args()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/async_llm_engine.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 2
number of parameters: 7

id: 171
unit: def build_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v3.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 6
number of parameters: 1

id: 172
unit: def save_checkpoint()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 6
number of parameters: 2

id: 173
unit: def timeit()
file: chatlearn/runtime/decorator.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 15
number of parameters: 2

id: 174
unit: def load_module()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 10
number of parameters: 2

id: 175
unit: def check_and_fuse_lora()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 9
number of parameters: 3

id: 176
unit: def check_and_unfuse_lora()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 9
number of parameters: 3

id: 177
unit: def save_checkpoint()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 5

id: 178
unit: def __init__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 9

id: 179
unit: def build_pipeline_layer_name_mapping()
file: chatlearn/utils/megatron_utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 12
number of parameters: 5

id: 180
unit: def listdict_to_dictlist()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 2

id: 181
unit: def setup_logger()
file: chatlearn/utils/logger.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 4
number of parameters: 3

id: 182
unit: def set_pipe_layer_num_offset()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 6
number of parameters: 1

id: 183
unit: def model_setup()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 4
number of parameters: 1

id: 184
unit: def offload_weights()
file: chatlearn/models/vllm/inference.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 7
number of parameters: 1

id: 185
unit: def load_state_dict()
file: chatlearn/models/vllm/vllm_model.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 7
number of parameters: 4

id: 186
unit: def load_state_dict()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/loader.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 7
number of parameters: 4

id: 187
unit: def load_state_dict()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/loader.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 7
number of parameters: 4

id: 188
unit: def offload_weights()
file: chatlearn/models/megatron/memory_manager/inference.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 7
number of parameters: 1

id: 189
unit: def model_setup()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 3
number of parameters: 1

id: 190
unit: def add_remote_func()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 10
number of parameters: 1

id: 191
unit: def register_func()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 1

id: 192
unit: def split_along_batch()
file: chatlearn/runtime/decorator.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 6
number of parameters: 2

id: 193
unit: def _setup_tensorboard()
file: chatlearn/schedule/metric_manager.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 1

id: 194
unit: def _alloc_primary_store()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 3

id: 195
unit: def __init__()
file: chatlearn/models/vllm/vllm_model.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 4

id: 196
unit: def free_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v4.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 7
number of parameters: 1

id: 197
unit: def setup_dataset()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 1

id: 198
unit: def logging_summary()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 2

id: 199
unit: def add_recv_actor_stage2()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 3

id: 200
unit: def sort_send_actors()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 7
number of parameters: 3

id: 201
unit: def start_ray_cluster()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 0

id: 202
unit: def __init__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 7

id: 203
unit: def split_batch()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 6
number of parameters: 1

id: 204
unit: def _convert_v1_inputs()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 3

id: 205
unit: def empty_cache()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 1

id: 206
unit: def __init__()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 8
number of parameters: 3

id: 207
unit: def _reset_metrics_stats_args()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 1

id: 208
unit: def convert_v1_inputs()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 3

id: 209
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 4

id: 210
unit: def reorder_actors()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 3

id: 211
unit: def set_dist_env()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 2

id: 212
unit: def get_next_data()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 6
number of parameters: 4

id: 213
unit: def mcts_loop()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 7

id: 214
unit: def sprl_loop()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 7

id: 215
unit: def logging_memory()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 6
number of parameters: 1

id: 216
unit: def monitor_error()
file: chatlearn/runtime/decorator.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 2

id: 217
unit: def concat_params()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 6
number of parameters: 2

id: 218
unit: def _clear_sync_send_recv_parameters()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 8
number of parameters: 2

id: 219
unit: def create_allgather_group()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 6
number of parameters: 3

id: 220
unit: def clear_cache()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 3

id: 221
unit: def warmup_groups()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 6
number of parameters: 1

id: 222
unit: def __init__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 8
number of parameters: 7

id: 223
unit: def iter_dynamic()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 1

id: 224
unit: def parse_args()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 0

id: 225
unit: def get_load_dir()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 226
unit: def copy_to_primary_store()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 2

id: 227
unit: def multi_thread_data_processing()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 4

id: 228
unit: def monitor()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 1

id: 229
unit: def __init__()
file: chatlearn/utils/logger.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 7

id: 230
unit: def __init__()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 3
number of parameters: 3

id: 231
unit: def broadcast_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 8
number of parameters: 5

id: 232
unit: def empty_cache()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 8
number of parameters: 1

id: 233
unit: def profile_cache_blocks()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 1

id: 234
unit: def execute_method()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/worker_base.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 4

id: 235
unit: def execute_method()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/worker_base.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 4

id: 236
unit: def _create_remote_models()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 3
number of parameters: 1

id: 237
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 5

id: 238
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 5

id: 239
unit: def _find_all_parents()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 3

id: 240
unit: def set_model_prefix()
file: chatlearn/synchronizer/megatron_megatron.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 7
number of parameters: 3

id: 241
unit: def validate_sync_results_parallel()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 9
number of parameters: 6

id: 242
unit: def in_dlc_env()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 0

id: 243
unit: def _delete_ckpt_files()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 8
number of parameters: 1

id: 244
unit: def __init__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 245
unit: def add_raw_batch()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 1

id: 246
unit: def __iter__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 1

id: 247
unit: def __str__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 9
number of parameters: 1

id: 248
unit: def split_attn_state()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 5

id: 249
unit: def broadcast_var_object_dict()
file: chatlearn/utils/dist_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 2

id: 250
unit: def save_hf_format()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 4

id: 251
unit: def load_stage_outputs()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 8
number of parameters: 3

id: 252
unit: def _setup_collective_group()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 5

id: 253
unit: def split_along_batch()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 2

id: 254
unit: def setup()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 255
unit: def concat_along_batch()
file: chatlearn/runtime/decorator.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 1

id: 256
unit: def parse()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 2

id: 257
unit: def recursive_print()
file: chatlearn/tools/megatron_to_hf.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 3

id: 258
unit: def setup_collective_group()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 6
number of parameters: 1

id: 259
unit: def init()
file: chatlearn/launcher/initialize.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 1

id: 260
unit: def __init__()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 4

id: 261
unit: def _to_dist_model()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 2

id: 262
unit: def cal_consumed_each()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 263
unit: def __repr__()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 1

id: 264
unit: def parse_args_from_yaml()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 265
unit: def __init__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 3

id: 266
unit: def check_nested_2_level_list()
file: chatlearn/utils/future.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 1

id: 267
unit: def split_index()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 268
unit: def map_reduce_metrics()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 1

id: 269
unit: def get_megatron_version()
file: chatlearn/utils/megatron_import_memory_helper.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 6
number of parameters: 0

id: 270
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 271
unit: def offload_main_weights()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 1

id: 272
unit: def get_ds_train_config()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 1

id: 273
unit: def setup()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 1

id: 274
unit: def fix_act_ordering()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 275
unit: def _calculate_max_workers()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 276
unit: def _get_total_gpu_required()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 7
number of parameters: 1

id: 277
unit: def set_func_decorator()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 278
unit: def set_dist_env_concurrent()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 279
unit: def clean()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 6
number of parameters: 1

id: 280
unit: def run()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 7
number of parameters: 2

id: 281
unit: def elapsed()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 3

id: 282
unit: def parse_args()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 2

id: 283
unit: def parse_function_return_num()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 1

id: 284
unit: def create_device_mesh()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 3

id: 285
unit: def ray_put_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 3

id: 286
unit: def ray_get_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 4

id: 287
unit: def free_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v3.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 1

id: 288
unit: def _ds_init_train_model()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 4

id: 289
unit: def create_actor()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 4

id: 290
unit: def setup_queues()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 1

id: 291
unit: def set_multiple_datasets()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 292
unit: def set_multiple_datasets()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 293
unit: def set_multiple_datasets()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 294
unit: def __init__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 3

id: 295
unit: def sync()
file: chatlearn/synchronizer/parameter_sync_fsdp.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 3

id: 296
unit: def add_routed_experts_regrouping_actor()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 3

id: 297
unit: def _sync_send_recv()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 6

id: 298
unit: def sync_parameters()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 5

id: 299
unit: def __iter__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 1

id: 300
unit: def __iter__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 8
number of parameters: 1

id: 301
unit: def __str__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 1

id: 302
unit: def detect_and_insert_code()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 6

id: 303
unit: def worker_dump_parameters()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 304
unit: def generate_vllm()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 5

id: 305
unit: def check_param_exists()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 306
unit: def onload_weights()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 307
unit: def onload_optimizer_states()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 2

id: 308
unit: def get_pipeline_stage_layer_num()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 1

id: 309
unit: def allgather_routed_expert_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 3

id: 310
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 2

id: 311
unit: def only_optimize_lora_parameters()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 9
number of parameters: 4

id: 312
unit: def empty_cache()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 1

id: 313
unit: def _create_actor()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 6

id: 314
unit: def batch_per_episode()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 315
unit: def next_batch()
file: chatlearn/runtime/trainer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 1

id: 316
unit: def _resume_from_data_checkpoint()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 1

id: 317
unit: def parallel_execute_collective_tasks()
file: chatlearn/synchronizer/scheduler.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 2

id: 318
unit: def set_send_actors_to_regroup_routed_experts()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 2

id: 319
unit: def clear_cache()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 3

id: 320
unit: def get_addr()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 0

id: 321
unit: def iter_internal()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 2

id: 322
unit: def batch_generation_ranking()
file: chatlearn/data/ranking.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 323
unit: def __init__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 4

id: 324
unit: def update_data_uid()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 3

id: 325
unit: def list_logs()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 326
unit: def reopen_if_necessary()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 327
unit: def _get_shard_range()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 328
unit: def execute()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 3

id: 329
unit: def empty_cache()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 1

id: 330
unit: def empty_cuda_graph()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 331
unit: def get_fsdp_param_name()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 332
unit: def offload_optimizer_states()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 7
number of parameters: 2

id: 333
unit: def parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 334
unit: def named_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 335
unit: def param_to_name()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 336
unit: def get_parameter_shape()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 7
number of parameters: 3

id: 337
unit: def alltoall_routed_expert_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 2

id: 338
unit: def _get_logits()
file: chatlearn/models/vllm/hooks/vllm_0_5_1/logits_processor.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 4

id: 339
unit: def _get_logits()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/logits_processor.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 4

id: 340
unit: def forward()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 341
unit: def print_trainable_parameters()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 342
unit: def _get_grad_to_params()
file: chatlearn/models/megatron/memory_manager/trainer_v4.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 2

id: 343
unit: def __init__()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 344
unit: def _ds_init_eval_model()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 345
unit: def __init__()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 346
unit: def execute()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 2

id: 347
unit: def evaluate()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 2

id: 348
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 3

id: 349
unit: def fix_shared_expert_ordering()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 2

id: 350
unit: def get_free_ports()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 7
number of parameters: 0

id: 351
unit: def execute_with_timeout()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 352
unit: def init_ray()
file: chatlearn/launcher/initialize.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 1

id: 353
unit: def __init__()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 6

id: 354
unit: def _get_checkpoint_path()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 355
unit: def resume()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 356
unit: def _wandb_scalar_dict()
file: chatlearn/schedule/metric_manager.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 4

id: 357
unit: def get_free_port()
file: chatlearn/schedule/port_manager.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 358
unit: def get()
file: chatlearn/utils/future.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 6
number of parameters: 1

id: 359
unit: def parse_function_args()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 1

id: 360
unit: def nvml_proc_memory_info()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 361
unit: def slice_by_index_along_batch()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 2

id: 362
unit: def get_weight_ipc_handles_by_name()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 2

id: 363
unit: def init()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 1

id: 364
unit: def set_cache_config()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 365
unit: def _get_logits()
file: chatlearn/models/vllm/hooks/vllm_0_3_0/sampler.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 4

id: 366
unit: def _get_grad_dtype_to_params()
file: chatlearn/models/megatron/memory_manager/trainer_v1v2.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 2

id: 367
unit: def update_models()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 2

id: 368
unit: def regroup_inqueue()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 4

id: 369
unit: def set_dataset()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 2

id: 370
unit: def set_dataset()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 2

id: 371
unit: def set_dataset()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 2

id: 372
unit: def get_input_queues()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 1

id: 373
unit: def add_dependent_colocate_model_results()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 6
number of parameters: 4

id: 374
unit: def get_or_cache()
file: chatlearn/synchronizer/base.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 5

id: 375
unit: def get_or_cache()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 5

id: 376
unit: def sync_broadcast()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 6

id: 377
unit: def sync_alltoall()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 4

id: 378
unit: def set_sync_param_names()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 7

id: 379
unit: def _single_thread_sync()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 5

id: 380
unit: def resume_meta()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 381
unit: def __init__()
file: chatlearn/schedule/metric_manager.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 2

id: 382
unit: def set_parameter_sync()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 383
unit: def _find_param_recv_models()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 7
number of parameters: 2

id: 384
unit: def __init__()
file: chatlearn/schedule/resource_manager.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 385
unit: def sort_fun()
file: chatlearn/data/ranking.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 386
unit: def __init__()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 387
unit: def get_model()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 388
unit: def get_megatron_sharded_states()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 4

id: 389
unit: def to_device()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 6
number of parameters: 2

id: 390
unit: def execute_in_parallel()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 391
unit: def map_metrics()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 392
unit: def reduce_metrics()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 393
unit: def unwrap_func()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 394
unit: def get_tokenizer()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 4

id: 395
unit: def get_train_ds_config()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 396
unit: def _set_sync_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 4

id: 397
unit: def send_recv_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 5

id: 398
unit: def save_stage_outputs()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 4

id: 399
unit: def _get_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v3.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 400
unit: def tensor_decomp_pg_loss()
file: chatlearn/models/megatron/ops/policy_gradient.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 7

id: 401
unit: def prepare()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 402
unit: def check_param_exists()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 403
unit: def batch_size()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 2

id: 404
unit: def decorate_class_func()
file: chatlearn/runtime/decorator.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 6
number of parameters: 5

id: 405
unit: def parse_expr()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 406
unit: def detect_and_insert_code()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 6
number of parameters: 5

id: 407
unit: def split_attn_state()
file: chatlearn/tools/megatron_to_hf.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 408
unit: def get_megatron_sharded_states()
file: chatlearn/tools/megatron_to_hf.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 4

id: 409
unit: def allgather_routed_experts()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 5

id: 410
unit: def alltoall_routed_experts()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 4

id: 411
unit: def destroy_collective_group()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 412
unit: def get_load_balance_dst_rank()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 413
unit: def sync_allgather()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 5

id: 414
unit: def _tensorboard_scalar_dict()
file: chatlearn/schedule/metric_manager.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 4

id: 415
unit: def repeat()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 3

id: 416
unit: def flatten()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 417
unit: def detect_and_insert_code_to_func()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 6

id: 418
unit: def get_use_legacy_models()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 419
unit: def add_extra_args()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 420
unit: def init_memory_manager()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 421
unit: def create_model()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 422
unit: def get_rng_state()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 423
unit: def get_local_param_ranks()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 424
unit: def enable_stage_resume()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 425
unit: def add_extra_args()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 426
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 427
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 428
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 429
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 430
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 431
unit: def load_weights()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 432
unit: def forward()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 433
unit: def __init__()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 434
unit: def _optimizer_load_state_bucket_into_device()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 435
unit: def __init__()
file: chatlearn/models/megatron/memory_manager/trainer_v3.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 436
unit: def __init__()
file: chatlearn/models/megatron/memory_manager/inference.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 437
unit: def __init__()
file: chatlearn/models/megatron/memory_manager/trainer_v1v2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 438
unit: def __init__()
file: chatlearn/models/megatron/memory_manager/trainer_v4.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 439
unit: def create_model()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 440
unit: def group_dist_actors_by_tp_rank()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 441
unit: def call_vllm_engine_and_workers_remote_funcs()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 4

id: 442
unit: def call_replica_serial_func()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 4

id: 443
unit: def get_all_merged_data()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 4

id: 444
unit: def eval()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 445
unit: def visit_func()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 6
number of parameters: 2

id: 446
unit: def wait_colocate_models_to_finish()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 3

id: 447
unit: def flatten()
file: chatlearn/synchronizer/parameter_sync_fsdp.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 448
unit: def check_param_names()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 5

id: 449
unit: def _multi_thread_sync_for_tp_num_mapping_gt_1()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 450
unit: def __init__()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 6

id: 451
unit: def _get_latest_iteration()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 452
unit: def log()
file: chatlearn/schedule/metric_manager.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 4

id: 453
unit: def place_models_to_remote_devices()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 8
number of parameters: 3

id: 454
unit: def get_start_end_idx()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 455
unit: def update_random_idxs()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 456
unit: def _get_batch()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 457
unit: def next()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 458
unit: def has_log_file_list_changed()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 2

id: 459
unit: def update_dict()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 2

id: 460
unit: def __init__()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 461
unit: def get_element_from_dict_by_path()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 462
unit: def coalesced_comm_dense()
file: chatlearn/utils/dist_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 4

id: 463
unit: def _get_total_numel()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 464
unit: def _alloc_gpu_buffer()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 465
unit: def get_return_lines()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 1

id: 466
unit: def get_return_value_num()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 467
unit: def get_indent_count()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 468
unit: def get_ray_status()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 0

id: 469
unit: def get_full_proc_memory_info()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 470
unit: def log_rank_0()
file: chatlearn/utils/logger.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 2

id: 471
unit: def debug_rank_0()
file: chatlearn/utils/logger.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 2

id: 472
unit: def run_vllm()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 473
unit: def megatron_model()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 474
unit: def build_pipeline_layer_name_mapping()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 5

id: 475
unit: def save_checkpoint()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 476
unit: def build_all_dataset()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 477
unit: def reset_vllm()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 478
unit: def init()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 479
unit: def onload_weights()
file: chatlearn/models/vllm/inference.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 480
unit: def from_engine_args()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/async_llm_engine.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 481
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 482
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 483
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 484
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 485
unit: def create_trainer_memory_manager()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 486
unit: def onload_main_weights()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 487
unit: def onload_weights()
file: chatlearn/models/megatron/memory_manager/inference.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 488
unit: def _setup_ranks()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 489
unit: def __init__()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 490
unit: def call_replica_func()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 4

id: 491
unit: def get_all_merged_data_list()
file: chatlearn/runtime/evaluator.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 492
unit: def _set_flow()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 493
unit: def __init__()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 494
unit: def data_parallel_size()
file: chatlearn/runtime/trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 495
unit: def get_indent_count()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 496
unit: def get_element_from_dict_by_path()
file: chatlearn/tools/megatron_to_hf.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 497
unit: def main()
file: chatlearn/tools/megatron_to_hf.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 498
unit: def map_name_from_src_to_dst()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 499
unit: def __init__()
file: chatlearn/synchronizer/parameter_sync_fsdp.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 500
unit: def setup_collective_group()
file: chatlearn/synchronizer/parameter_sync_fsdp.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 501
unit: def set_sync_param_names_stage2()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 7

id: 502
unit: def get_iter_keys()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 503
unit: def batch_shuffle()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 504
unit: def get_and_clear_metrics()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 505
unit: def get()
file: chatlearn/data/storage.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 506
unit: def __init__()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 507
unit: def should_update_filenames()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 508
unit: def is_proc_alive()
file: chatlearn/utils/log_monitor.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 509
unit: def start()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 510
unit: def print_rank_0()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 511
unit: def initialize_megatron()
file: chatlearn/utils/megatron_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 512
unit: def finish_mpu_init()
file: chatlearn/utils/megatron_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 0

id: 513
unit: def release_gpu_buffer()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 514
unit: def __init__()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 515
unit: def get_or_cache()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 5

id: 516
unit: def display()
file: chatlearn/utils/logger.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 517
unit: def set_global_variables()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 518
unit: def model()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 519
unit: def offload_for_workers()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 5

id: 520
unit: def onload_for_workers()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 5

id: 521
unit: def clear_cache()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 522
unit: def expert_model_parallel_size()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 523
unit: def get_parameter_names()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 2

id: 524
unit: def get_and_clear_metrics()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 525
unit: def setup()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 526
unit: def reinit_cache_engine()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 527
unit: def clear_cache()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 528
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 529
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 530
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 531
unit: def fuse_lora_layer()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

id: 532
unit: def unfuse_lora_layer()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

id: 533
unit: def recursive_getattr()
file: chatlearn/models/megatron/lora/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 534
unit: def recursive_setattr()
file: chatlearn/models/megatron/lora/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 535
unit: def _flat_param_groups()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 536
unit: def set_env()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 537
unit: def _get_if_not_none()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 538
unit: def add_remote_func()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 539
unit: def call_remote_funcs()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 540
unit: def get_and_clear_metrics()
file: chatlearn/runtime/evaluator.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 541
unit: def setup()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 542
unit: def _next_model()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 543
unit: def __init__()
file: chatlearn/runtime/trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 544
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 545
unit: def _create_remote_models()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 546
unit: def repair_loader_model_provider()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 547
unit: def repair_loader_mcore_model_provider()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 548
unit: def transform_parameters()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 549
unit: def _get_dst_name()
file: chatlearn/synchronizer/megatron_megatron.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 550
unit: def is_same_gpu()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 551
unit: def setup_rank_mapping()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 552
unit: def _clear_sorted_send_actors()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 553
unit: def check_and_destroy_collective_group()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 554
unit: def get_placement_group_state()
file: chatlearn/schedule/resource_manager.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 555
unit: def remove_placement_groups()
file: chatlearn/schedule/resource_manager.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 556
unit: def get_and_clear_metrics()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 557
unit: def shuffle()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 558
unit: def get_path()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 559
unit: def convert_type()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 560
unit: def __init__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 561
unit: def get()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 562
unit: def get()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 563
unit: def write()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 6

id: 564
unit: def __init__()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 565
unit: def initialize_vllm()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 566
unit: def _unpin()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 567
unit: def __init__()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 568
unit: def _link_tensor_data_to_gpu_buffer()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 569
unit: def copy_to_gpu_buffer()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 570
unit: def get_free_port()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 571
unit: def dict_to_simplenamespace()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 572
unit: def create_optimizer()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 573
unit: def _z3_params_to_fetch()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 1

id: 574
unit: def init()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 575
unit: def setup()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 576
unit: def update_weights_from_ipc_handles()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 577
unit: def reinit_cache_engine()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 578
unit: def init_fn()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 579
unit: def setup_distributed()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 580
unit: def peak_memory()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 581
unit: def load_rng_state()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 582
unit: def next_batch()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 583
unit: def setup_collective_group()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 5

id: 584
unit: def set_sync_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 4

id: 585
unit: def get_stage_outputs_path()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 586
unit: def __init__()
file: chatlearn/models/vllm/inference.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 587
unit: def is_vllm_v2()
file: chatlearn/models/vllm/__init__.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 588
unit: def load_weights()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/loader.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 589
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 590
unit: def __init__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 591
unit: def _prompt_to_llm_inputs()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/input_preprocess.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 592
unit: def load_weights()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/loader.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 593
unit: def from_engine_args()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/llm_engine.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 594
unit: def forward()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 595
unit: def state_dict()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 5

id: 596
unit: def fuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 597
unit: def unfuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 598
unit: def fuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 599
unit: def unfuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 600
unit: def fuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 601
unit: def unfuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 602
unit: def fuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 603
unit: def unfuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 604
unit: def fuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 605
unit: def unfuse_lora_weight()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 606
unit: def get_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v1v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 607
unit: def limit_min_prompt_length()
file: chatlearn/models/megatron/hooks/generation.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 608
unit: def set_seed()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 609
unit: def setup_distributed()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 610
unit: def get_ds_eval_config()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 611
unit: def save_checkpoint()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 612
unit: def get_dist_env()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 613
unit: def call_actor_remote_func()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 5

id: 614
unit: def call_vllm_engine_remote_funcs()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 615
unit: def init()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 616
unit: def __init__()
file: chatlearn/runtime/evaluator.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 617
unit: def __init__()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 618
unit: def dump_parameters()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 619
unit: def parse_assign_target()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 620
unit: def add_input_node()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 621
unit: def __init__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 622
unit: def find_module()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 623
unit: def __init__()
file: chatlearn/synchronizer/base.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 624
unit: def __init__()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 625
unit: def map_src_to_dst()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 626
unit: def map_name_from_src_to_dst()
file: chatlearn/synchronizer/megatron_megatron.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 5

id: 627
unit: def sync_send_recv()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 6

id: 628
unit: def recover_synchronizer()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 629
unit: def reset_synchronizer()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 630
unit: def get_master_addr()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 631
unit: def stop()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 632
unit: def _make_checkpoint_path()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 633
unit: def create_from_type()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 634
unit: def shuffle()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 635
unit: def __iter__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 636
unit: def __init__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 637
unit: def __setattr__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 638
unit: def validate()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 639
unit: def stop()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 640
unit: def reset()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 641
unit: def __call__()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 642
unit: def dst_names()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 643
unit: def finish_mpu_init()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 0

id: 644
unit: def ensure_directory_exists()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 645
unit: def update_layer_num()
file: chatlearn/utils/megatron_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 646
unit: def _pin()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 647
unit: def get_host_addr()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 648
unit: def __init__()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 649
unit: def __init__()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 650
unit: def set()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 651
unit: def logger()
file: chatlearn/utils/logger.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 652
unit: def is_initialized()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 653
unit: def set_dependencies()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 654
unit: def get_eval_ds_config()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 655
unit: def get_optimizer_grouped_parameters()
file: chatlearn/models/deepspeed/deepspeed_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 7

id: 656
unit: def expert_parallel_rank()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 657
unit: def get_pipeline_stage_layer_offset()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 658
unit: def data_ckpt_manager()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 659
unit: def save_data_checkpoint()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 660
unit: def broadcast_dummy_tensor_send()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 661
unit: def broadcast_dummy_tensor_recv()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 662
unit: def destroy_collective_group()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 663
unit: def set_recv_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 664
unit: def get_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 665
unit: def timers()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 666
unit: def model_provider()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 667
unit: def _reset_scheduler()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 668
unit: def _add_request()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 669
unit: def load_weights()
file: chatlearn/models/vllm/vllm_model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 670
unit: def compute_logits()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 671
unit: def sample()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 672
unit: def compute_logits()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 673
unit: def sample()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 674
unit: def apply_rotary_pos_emb_variable_seq()
file: chatlearn/models/megatron/hooks/transformer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 675
unit: def is_last_rank()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 676
unit: def get_actor()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 2

id: 677
unit: def split_list()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 678
unit: def __init__()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 679
unit: def setup()
file: chatlearn/runtime/trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 680
unit: def __init__()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 681
unit: def setup()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 682
unit: def decode_data()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 683
unit: def parse_assign()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 684
unit: def __init__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 685
unit: def repair_loader_put_reward()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 686
unit: def repair_saver_get_reward()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 687
unit: def repair_import_utils()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 688
unit: def repair_saver_mcore_import_error()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 689
unit: def repair_saver_mcore_get_reward()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 690
unit: def num_src_pipeline_stage()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 691
unit: def num_dst_pipeline_stage()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 692
unit: def num_src_tensor_parallel()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 693
unit: def num_dst_tensor_parallel()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 694
unit: def num_src_expert_parallel()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 695
unit: def num_dst_expert_parallel()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 696
unit: def get_actor_pipe_rank()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 697
unit: def get_actor_tp_rank()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 698
unit: def get_actor_ep_rank()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 699
unit: def get_actor_dp_rank()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 700
unit: def check_and_setup_collective_group()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 701
unit: def num_src_hyper_expert_parallel()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 702
unit: def num_dst_hyper_expert_parallel()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 703
unit: def get_actor_hep_rank()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 704
unit: def filter_known_msg()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 705
unit: def _get_checkpoint_path_name()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 706
unit: def validate()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 707
unit: def start_error_monitor()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 708
unit: def __init__()
file: chatlearn/schedule/port_manager.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 709
unit: def get_start_end_idx()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 710
unit: def sync()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 711
unit: def put()
file: chatlearn/data/storage.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 712
unit: def __init__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 713
unit: def __init__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 714
unit: def cuda_sync()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 715
unit: def get_dst_name()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 716
unit: def __del__()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 4
number of parameters: 1

id: 717
unit: def copy_to_primary_store()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 718
unit: def release_gpu_buffer()
file: chatlearn/utils/flat_tensors.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 719
unit: def get_attributes()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 4
number of parameters: 1

id: 720
unit: def is_connection_refused()
file: chatlearn/utils/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 721
unit: def check_megatron_versions()
file: chatlearn/utils/megatron_import_memory_helper.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 722
unit: def set_address()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 723
unit: def get_args()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 724
unit: def is_decorated()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 725
unit: def set_wrap_func()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 726
unit: def reset_dependencies()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 727
unit: def set_vllm_actors()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 728
unit: def dump_parameters()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 729
unit: def peak_memory()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 730
unit: def offload_weights()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 731
unit: def onload_weights()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 732
unit: def offload_optimizer_states()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 733
unit: def onload_optimizer_states()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 734
unit: def offload_main_weights()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 735
unit: def onload_main_weights()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 736
unit: def offload_weights()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 737
unit: def onload_weights()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 738
unit: def free_grad_buffers()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 739
unit: def build_grad_buffers()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 740
unit: def get()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 741
unit: def reset_eval_data_iter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 742
unit: def fuse_lora_layer()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 743
unit: def unfuse_lora_layer()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 744
unit: def reset_sync_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 745
unit: def clear_send_recv_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 746
unit: def clear_sync_send_recv_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 747
unit: def timer_summary()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 748
unit: def offload_weights()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 749
unit: def onload_weights()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 750
unit: def execute_model()
file: chatlearn/models/vllm/hooks/vllm_0_5_1/worker.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 751
unit: def _get_device_name()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/format_device_name.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 752
unit: def inspect_model_cls()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/registry.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 753
unit: def reset_parameters()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 754
unit: def reset_parameters()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 755
unit: def reset_parameters()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 756
unit: def reset_parameters()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 757
unit: def reset_parameters()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 758
unit: def modify_code()
file: chatlearn/models/megatron/hooks/transformer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 759
unit: def peak_memory()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 760
unit: def _get_func_args()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 761
unit: def terminate()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 762
unit: def __init__()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 763
unit: def create_engine_actor()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 764
unit: def add_replica()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 765
unit: def group_dist_actors_by_tp_rank()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 766
unit: def enable_offload()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 767
unit: def get_merged_data_locked()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 6

id: 768
unit: def num_iteration()
file: chatlearn/runtime/trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 2

id: 769
unit: def timer_summary()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 770
unit: def before_episode()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 771
unit: def after_episode()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 772
unit: def stop()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 773
unit: def set_parameter_sync()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 774
unit: def set_trainer()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 775
unit: def set_environment()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 776
unit: def set_evaluator()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 777
unit: def __enter__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 778
unit: def __init__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 779
unit: def repair_entry_file()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 780
unit: def repair_mcore_block_key()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 781
unit: def exist_checkpoint_util()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 782
unit: def repair_loader_llama_mistral()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 783
unit: def __init__()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 784
unit: def __init__()
file: chatlearn/__init__.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 785
unit: def __init__()
file: chatlearn/__init__.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 786
unit: def __init__()
file: chatlearn/__init__.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 787
unit: def __init__()
file: chatlearn/__init__.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 788
unit: def __init__()
file: chatlearn/synchronizer/scheduler.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 789
unit: def regroup_params_to_sync()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 5

id: 790
unit: def map_src_to_dst()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 791
unit: def map_src_to_dst()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 792
unit: def insert_actor2rank()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 793
unit: def insert_actor2model()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 794
unit: def sync_allgather_multi_threads()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 795
unit: def _multi_thread_sync_for_tp_num_mapping_eq_1()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 796
unit: def routed_experts_filter()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 2

id: 797
unit: def params_except_routed_expert_filter()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 2

id: 798
unit: def get_dlc_env()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 799
unit: def _set_latest_iteration()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 800
unit: def stop()
file: chatlearn/schedule/metric_manager.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 801
unit: def warmup_collective_topology()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 802
unit: def _set_dist_env()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 803
unit: def __len__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 804
unit: def __len__()
file: chatlearn/data/sampler.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 805
unit: def has_next()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 806
unit: def episode_relay_buffers()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 807
unit: def total_samples()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 808
unit: def batch_per_episode()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 809
unit: def queue_not_empty()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 810
unit: def get_samples()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 811
unit: def __len__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 812
unit: def buffer()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 813
unit: def episode_id()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 814
unit: def __call__()
file: chatlearn/data/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 815
unit: def __init__()
file: chatlearn/data/storage.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 816
unit: def __init__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 817
unit: def __repr__()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 818
unit: def validate()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 819
unit: def is_changed()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 820
unit: def active_module_args()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 821
unit: def active_module_args()
file: chatlearn/utils/arguments.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 822
unit: def __init__()
file: chatlearn/utils/timer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 823
unit: def embedding_sync_map()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 824
unit: def layer_sync_map()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 825
unit: def final_layer_sync_map()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 826
unit: def concat_params_dict()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 827
unit: def to_fix_shared_expert_ordering()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 828
unit: def to_allgather_roututed_experts_dict()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 829
unit: def to_alltoall_roututed_experts_dict()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 830
unit: def to_fix_act_ordering_dict()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 831
unit: def to_fix_qkv_ordering_dict()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 832
unit: def map_src_to_dst()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 833
unit: def fix_qwen_query_key_value_ordering()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 834
unit: def load_checkpoint()
file: chatlearn/utils/vllm_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 6

id: 835
unit: def is_set()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 836
unit: def error_msg()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 837
unit: def error_address()
file: chatlearn/utils/error_monitor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 838
unit: def _ensure_var_is_initialized()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 839
unit: def set_initialized()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 840
unit: def set_decorated()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 841
unit: def get_dependencies()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 842
unit: def get_vllm_actors()
file: chatlearn/utils/global_vars.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 843
unit: def is_last_rank()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 844
unit: def num_layers()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 845
unit: def data_parallel_size()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 846
unit: def data_parallel_rank()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 847
unit: def tensor_parallel_rank()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 848
unit: def pipeline_parallel_rank()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 849
unit: def tensor_model_parallel_size()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 850
unit: def expert_model_parallel_size()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 851
unit: def tensor_and_expert_model_parallel_size()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 852
unit: def model_setup_for_workers()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 853
unit: def empty_cache_for_workers()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 854
unit: def empty_cuda_graph_for_workers()
file: chatlearn/models/vllm_module_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 855
unit: def get_visible_gpus()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 856
unit: def data_parallel_size()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 857
unit: def data_parallel_rank()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 858
unit: def tensor_parallel_rank()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 859
unit: def pipeline_parallel_rank()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 860
unit: def expert_model_parallel_size()
file: chatlearn/models/fsdp_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 861
unit: def add_extra_args()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 862
unit: def megatron_args()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 863
unit: def pipeline_model_parallel_size()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 864
unit: def tensor_model_parallel_size()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 865
unit: def tensor_and_expert_model_parallel_size()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 866
unit: def data_parallel_size()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 867
unit: def data_parallel_rank()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 868
unit: def pipeline_parallel_rank()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 869
unit: def tensor_parallel_rank()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 870
unit: def tensor_and_expert_parallel_group()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 871
unit: def num_layers()
file: chatlearn/models/megatron_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 872
unit: def get_sync_buffer()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 873
unit: def set_tp_num_mapping()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 874
unit: def tp_num_mapping()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 875
unit: def set_buffer_num()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 876
unit: def get_buffer_num()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 877
unit: def set_tp_division()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 878
unit: def get_tp_division()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 879
unit: def is_colocate()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 880
unit: def set_colocate()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 881
unit: def finalize()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 882
unit: def _assert_not_finalized()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 883
unit: def get_runtime_args()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 884
unit: def runtime_args()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 885
unit: def model_args()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 886
unit: def module_args()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 887
unit: def parameter_sync_frequency()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 888
unit: def set_error_signal()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 889
unit: def error()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 890
unit: def put()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 891
unit: def after_episode()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 892
unit: def num_replica()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 893
unit: def num_gpu_per_replica()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 894
unit: def rank()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 895
unit: def get_rank()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 896
unit: def is_last_rank()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 897
unit: def set_send_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 898
unit: def clear_sync_parameters()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 899
unit: def get_parameter_to_sync_names()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 900
unit: def exist_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 901
unit: def parameter_shape()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 902
unit: def send_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 903
unit: def recv_parameter()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 904
unit: def pipeline_model_parallel_size()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 905
unit: def tensor_model_parallel_size()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 906
unit: def expert_model_parallel_size()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 907
unit: def set_storage()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 908
unit: def add_padding_config()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 909
unit: def padding_config()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 910
unit: def peak_memory()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 911
unit: def resume_training()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 912
unit: def get_address()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 913
unit: def is_master_node()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 914
unit: def set_src_parameter_model()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 915
unit: def src_parameter_model()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 916
unit: def onload()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 917
unit: def offload()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 918
unit: def world_size()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 919
unit: def get_data_parallel_rank()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 920
unit: def get_data_parallel_size()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 921
unit: def get_pipeline_stage_layer_num()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 922
unit: def get_pipeline_stage_layer_offset()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 923
unit: def set_synchronizer()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 924
unit: def expert_parallel_rank()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 925
unit: def pipeline_model_parallel_size()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 926
unit: def tensor_model_parallel_size()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 927
unit: def tensor_and_expert_model_parallel_size()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 928
unit: def data_parallel_size()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 929
unit: def data_parallel_rank()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 930
unit: def tensor_parallel_rank()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 931
unit: def pipeline_parallel_rank()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 932
unit: def num_layers()
file: chatlearn/models/vllm_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 933
unit: def forward()
file: chatlearn/models/vllm/vllm_model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 934
unit: def __reduce__()
file: chatlearn/models/vllm/hooks/vllm_0_5_1/llm_engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 935
unit: def init()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 936
unit: def __reduce__()
file: chatlearn/models/vllm/hooks/vllm_0_6_3/llm_engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 937
unit: def get_input_embeddings()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 938
unit: def get_input_embeddings()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3_moe.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 939
unit: def init()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/loader.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 940
unit: def __reduce__()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/llm_engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 941
unit: def get_input_embeddings()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 942
unit: def get_input_embeddings()
file: chatlearn/models/vllm/hooks/vllm_0_6_6/qwen3.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 943
unit: def distributed_kaiming_uniform_()
file: chatlearn/models/megatron/lora/initializer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 944
unit: def eval()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 945
unit: def train()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 946
unit: def eval()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 947
unit: def train()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 948
unit: def eval()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 949
unit: def train()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 950
unit: def eval()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 951
unit: def train()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 952
unit: def eval()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 953
unit: def train()
file: chatlearn/models/megatron/lora/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 954
unit: def offload_optimizer_states()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 955
unit: def onload_optimizer_states()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 956
unit: def get_grad_buffers()
file: chatlearn/models/megatron/memory_manager/trainer_v4.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 957
unit: def data_parallel_size()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 958
unit: def data_parallel_rank()
file: chatlearn/models/deepspeed_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 959
unit: def __init__()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 960
unit: def get_visible_gpus()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 961
unit: def world_size()
file: chatlearn/models/torch_module.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 962
unit: def module_args()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 963
unit: def runtime_args()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 964
unit: def master()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 965
unit: def tailer()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 966
unit: def actor_num()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 967
unit: def preprocess_actors()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 968
unit: def create_actor()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 969
unit: def placement_group()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 970
unit: def placement_group()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 971
unit: def set_dist_env()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 972
unit: def __str__()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 973
unit: def __repr__()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 974
unit: def master()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 975
unit: def peak_memory()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 976
unit: def trainable()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 977
unit: def module_args()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 978
unit: def actor_num()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 979
unit: def num_replica()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 980
unit: def total_gpu()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 981
unit: def total_cpu()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 982
unit: def num_gpu_per_replica()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 983
unit: def gpu_per_process()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 984
unit: def is_colocate()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 985
unit: def is_colocate()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 986
unit: def set_colocate_models()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 987
unit: def colocate_with()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 988
unit: def colocate_models()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 989
unit: def all_ranks()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 990
unit: def use_vllm_backend()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 991
unit: def __str__()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 992
unit: def __repr__()
file: chatlearn/runtime/dist_actor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 993
unit: def sample_per_episode()
file: chatlearn/runtime/evaluator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 994
unit: def post_process()
file: chatlearn/runtime/evaluator.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 995
unit: def set_timers()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 996
unit: def timers()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 997
unit: def first_node()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 998
unit: def first_model()
file: chatlearn/runtime/executor.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 999
unit: def sample_per_episode()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1000
unit: def make_experiences()
file: chatlearn/runtime/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1001
unit: def set_data_loader()
file: chatlearn/runtime/trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1002
unit: def set_timers()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1003
unit: def timers()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1004
unit: def _create_metric_manager()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1005
unit: def models()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1006
unit: def get_model()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1007
unit: def set_relay_sample_manager()
file: chatlearn/runtime/engine.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1008
unit: def encode_data()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1009
unit: def __init__()
file: chatlearn/runtime/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1010
unit: def __exit__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 1011
unit: def add_output_node()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1012
unit: def set_out_queues()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1013
unit: def set_input_queue()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1014
unit: def __str__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1015
unit: def __repr__()
file: chatlearn/runtime/model_flow.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1016
unit: def repair_saver_model_provider()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1017
unit: def repair_loader_mcore_import_error()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1018
unit: def repair_saver_mcore_model_provider()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1019
unit: def repair_loader_mcore_put_reward()
file: chatlearn/tools/megatron_checkpoint_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1020
unit: def map_name_from_src_to_dst()
file: chatlearn/synchronizer/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 5

id: 1021
unit: def allgather_routed_experts()
file: chatlearn/synchronizer/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 5

id: 1022
unit: def alltoall_routed_experts()
file: chatlearn/synchronizer/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 1023
unit: def transform_parameters()
file: chatlearn/synchronizer/base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1024
unit: def get_group_name()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 1025
unit: def frequency()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1026
unit: def empty_add_recv_actor()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1027
unit: def warmup_groups()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1028
unit: def build_rank_mapping_for_ep()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 1029
unit: def _clear_send_recv_param_names()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1030
unit: def sync_alltoall_multi_threads()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1031
unit: def build_rank_mapping_for_routed_experts()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1032
unit: def build_rank_mapping_for_params_except_routed_expert()
file: chatlearn/synchronizer/parameter_sync.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1033
unit: def is_local()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1034
unit: def get_job_name()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1035
unit: def get_rank()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1036
unit: def __init__()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1037
unit: def notify()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1038
unit: def add_error_node_and_msg()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1039
unit: def get_error_node_and_msg()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 1040
unit: def get_error_msg()
file: chatlearn/launcher/dlc_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1041
unit: def patch_ray()
file: chatlearn/launcher/initialize.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 1042
unit: def path_exists()
file: chatlearn/checkpoint/checkpoint_manager.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 1043
unit: def _get_group_name()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 1044
unit: def _get_sync_frequency()
file: chatlearn/schedule/model_manager.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 1045
unit: def set_env()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 2

id: 1046
unit: def init()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1047
unit: def setup()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1048
unit: def forward_step()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 3

id: 1049
unit: def train_step()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 3

id: 1050
unit: def eval_step()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 2

id: 1051
unit: def save_checkpoint()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 2

id: 1052
unit: def validate()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1053
unit: def before_episode()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1054
unit: def build_dataset()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 3

id: 1055
unit: def get_local_param_ranks()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1056
unit: def num_layers()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1057
unit: def offload_optimizer_states()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1058
unit: def onload_optimizer_states()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1059
unit: def offload_main_weights()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1060
unit: def onload_main_weights()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1061
unit: def offload_weights()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1062
unit: def onload_weights()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1063
unit: def free_grad_buffers()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1064
unit: def build_grad_buffers()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1065
unit: def data_parallel_size()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1066
unit: def data_parallel_rank()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1067
unit: def empty_cache()
file: chatlearn/models/base_module.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1068
unit: def offload_weights()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1069
unit: def onload_weights()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1070
unit: def free_grad_buffers()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1071
unit: def build_grad_buffers()
file: chatlearn/models/megatron/memory_manager/base_trainer.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 1

id: 1072
unit: def map_src_to_dst()
file: chatlearn/synchronizer/megatron_vllm.py
start line: 0
end line: 0
size: 1 LOC
McCabe index: 1
number of parameters: 3