id: 1
unit: def main()
file: scripts/generate_zen_dataset.py
start line: 0
end line: 0
size: 592 LOC
McCabe index: 14
number of parameters: 3

id: 2
unit: def train()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 300 LOC
McCabe index: 32
number of parameters: 1

id: 3
unit: def train()
file: trl/trainer/rloo_trainer.py
start line: 0
end line: 0
size: 249 LOC
McCabe index: 32
number of parameters: 1

id: 4
unit: def _compute_loss_liger()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 186 LOC
McCabe index: 36
number of parameters: 5

id: 5
unit: def from_pretrained()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 165 LOC
McCabe index: 54
number of parameters: 4

id: 6
unit: def main()
file: trl/scripts/vllm_serve.py
start line: 0
end line: 0
size: 124 LOC
McCabe index: 38
number of parameters: 1

id: 7
unit: def tokenize_row()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 112 LOC
McCabe index: 40
number of parameters: 4

id: 8
unit: def main()
file: scripts/log_example_reports.py
start line: 0
end line: 0
size: 111 LOC
McCabe index: 15
number of parameters: 2

id: 9
unit: def tokenize_row()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 103 LOC
McCabe index: 34
number of parameters: 4

id: 10
unit: def _compute_loss_liger()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 102 LOC
McCabe index: 13
number of parameters: 3

id: 11
unit: def main()
file: scripts/generate_toolcall_dataset.py
start line: 0
end line: 0
size: 90 LOC
McCabe index: 2
number of parameters: 3

id: 12
unit: def torch_call()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 87 LOC
McCabe index: 23
number of parameters: 5

id: 13
unit: def _process_tokens()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 86 LOC
McCabe index: 29
number of parameters: 4

id: 14
unit: def _process_tokens()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 86 LOC
McCabe index: 31
number of parameters: 4

id: 15
unit: def main()
file: scripts/log_reports.py
start line: 0
end line: 0
size: 83 LOC
McCabe index: 11
number of parameters: 1

id: 16
unit: def compute_reference_log_probs()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 70 LOC
McCabe index: 9
number of parameters: 2

id: 17
unit: def __call__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 68 LOC
McCabe index: 12
number of parameters: 3

id: 18
unit: def generate_completions()
file: trl/trainer/rloo_trainer.py
start line: 0
end line: 0
size: 68 LOC
McCabe index: 10
number of parameters: 2

id: 19
unit: def _compute_loss()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 66 LOC
McCabe index: 17
number of parameters: 3

id: 20
unit: def step()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 62 LOC
McCabe index: 18
number of parameters: 3

id: 21
unit: def unpack_tensor_with_streams()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 16
number of parameters: 1

id: 22
unit: def main()
file: trl/cli.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 13
number of parameters: 0

id: 23
unit: def generate_completions()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 9
number of parameters: 2

id: 24
unit: def torch_call()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 24
number of parameters: 5

id: 25
unit: def main()
file: trl/scripts/dpo.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 14
number of parameters: 3

id: 26
unit: def on_train_begin()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 54 LOC
McCabe index: 13
number of parameters: 5

id: 27
unit: def __call__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 25
number of parameters: 3

id: 28
unit: def __call__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 10
number of parameters: 3

id: 29
unit: def on_evaluate()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 52 LOC
McCabe index: 13
number of parameters: 5

id: 30
unit: def _pack_ffd()
file: trl/data_utils.py
start line: 0
end line: 0
size: 49 LOC
McCabe index: 18
number of parameters: 2

id: 31
unit: def __post_init__()
file: trl/trainer/grpo_config.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 18
number of parameters: 1

id: 32
unit: def tokenize()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 10
number of parameters: 4

id: 33
unit: def _generate_samples()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 3
number of parameters: 3

id: 34
unit: def __iter__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 14
number of parameters: 1

id: 35
unit: def pack_tensor()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 20
number of parameters: 1

id: 36
unit: def main()
file: trl/scripts/grpo.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 11
number of parameters: 3

id: 37
unit: def _move_model_to_vllm()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 29
number of parameters: 1

id: 38
unit: def step()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 12
number of parameters: 3

id: 39
unit: def _generate_samples()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 41 LOC
McCabe index: 5
number of parameters: 4

id: 40
unit: def print_env()
file: trl/scripts/env.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 20
number of parameters: 0

id: 41
unit: def _train_batched_samples()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 9
number of parameters: 5

id: 42
unit: def _calculate_rewards()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 19
number of parameters: 5

id: 43
unit: def main()
file: trl/scripts/sft.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 11
number of parameters: 3

id: 44
unit: def generate_from_model_and_ref()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 5
number of parameters: 4

id: 45
unit: def generate_from_model_and_ref()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 5
number of parameters: 4

id: 46
unit: def generate_from_model_and_ref()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 5
number of parameters: 4

id: 47
unit: def create_merge_config_ties()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 2
number of parameters: 1

id: 48
unit: def create_merge_config_dare_ties()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 2
number of parameters: 1

id: 49
unit: def calculate_loss()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 2
number of parameters: 7

id: 50
unit: def compute_reference_log_probs()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 4
number of parameters: 2

id: 51
unit: def get_eval_dataloader()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 36 LOC
McCabe index: 12
number of parameters: 3

id: 52
unit: def on_step_end()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 8
number of parameters: 5

id: 53
unit: def post_init()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 11
number of parameters: 2

id: 54
unit: def get_eval_dataloader()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 12
number of parameters: 2

id: 55
unit: def _prepare_peft_model()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 21
number of parameters: 4

id: 56
unit: def post_init()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 11
number of parameters: 2

id: 57
unit: def torch_call()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 21
number of parameters: 5

id: 58
unit: def get_eval_dataloader()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 11
number of parameters: 2

id: 59
unit: def log()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 13
number of parameters: 4

id: 60
unit: def create_merge_config_slerp()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 4
number of parameters: 1

id: 61
unit: def _generate_completions()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 12
number of parameters: 3

id: 62
unit: def _compute_judge()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 9
number of parameters: 4

id: 63
unit: def process_row()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 10
number of parameters: 5

id: 64
unit: def get_train_dataloader()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 5
number of parameters: 1

id: 65
unit: def __init__()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 6
number of parameters: 2

id: 66
unit: def compute_loss()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 17
number of parameters: 5

id: 67
unit: def main()
file: trl/scripts/kto.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 7
number of parameters: 3

id: 68
unit: def get_train_dataloader()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 6
number of parameters: 1

id: 69
unit: def _compute_judge()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 10
number of parameters: 4

id: 70
unit: def get_train_dataloader()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 7
number of parameters: 1

id: 71
unit: def visualize_samples()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 11
number of parameters: 2

id: 72
unit: def __init__()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 3
number of parameters: 5

id: 73
unit: def step()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 8
number of parameters: 2

id: 74
unit: def _generate_vllm()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 16
number of parameters: 3

id: 75
unit: def process_log_file()
file: scripts/log_reports.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 8
number of parameters: 1

id: 76
unit: def pipeline_step_with_grad()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 1
number of parameters: 13

id: 77
unit: def _config_check()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 4
number of parameters: 1

id: 78
unit: def get_eval_dataloader()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 10
number of parameters: 2

id: 79
unit: def _compute_kl_logps()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 3
number of parameters: 3

id: 80
unit: def prepare_deepspeed()
file: trl/models/utils.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 9
number of parameters: 2

id: 81
unit: def prepare_model_inputs()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 12
number of parameters: 4

id: 82
unit: def compute_liger_loss()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 7
number of parameters: 3

id: 83
unit: def build_tokenized_answer()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 5
number of parameters: 3

id: 84
unit: def build_tokenized_answer()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 5
number of parameters: 3

id: 85
unit: def compute_loss()
file: trl/trainer/gkd_trainer.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 3
number of parameters: 5

id: 86
unit: def _generate_completions()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 4
number of parameters: 3

id: 87
unit: def _compute_logprobs()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 5
number of parameters: 5

id: 88
unit: def prepare_fsdp()
file: trl/models/utils.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 6
number of parameters: 2

id: 89
unit: def show_tokens()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 6
number of parameters: 3

id: 90
unit: def _get_sample_prompt_embeddings()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 2
number of parameters: 3

id: 91
unit: def tokenize_row()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 8
number of parameters: 5

id: 92
unit: def _sync_fsdp_params_to_vllm()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 12
number of parameters: 4

id: 93
unit: def compute_accuracy()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 14
number of parameters: 1

id: 94
unit: def check_server()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 3

id: 95
unit: def __init__()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 23

id: 96
unit: def _process_completions()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 4

id: 97
unit: def judge()
file: trl/trainer/judges.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 4

id: 98
unit: def _generate()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 6
number of parameters: 3

id: 99
unit: def _process_completions()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 1
number of parameters: 4

id: 100
unit: def convert_state_dict_to_diffusers()
file: trl/models/sd_utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 10
number of parameters: 3

id: 101
unit: def pipeline_step()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 0

id: 102
unit: def get_train_dataloader()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 1

id: 103
unit: def get_peft_config()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 3
number of parameters: 1

id: 104
unit: def _split_kwargs()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 7
number of parameters: 2

id: 105
unit: def compute_reward_score()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 4

id: 106
unit: def instructions_formatting_function()
file: trl/extras/dataset_formatting.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 3
number of parameters: 1

id: 107
unit: def run()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 11
number of parameters: 3

id: 108
unit: def log()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 11
number of parameters: 4

id: 109
unit: def __init__()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 23

id: 110
unit: def log()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 11
number of parameters: 4

id: 111
unit: def __init__()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 31

id: 112
unit: def __init__()
file: trl/trainer/prm_trainer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 21

id: 113
unit: def __init__()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 21

id: 114
unit: def task_end_check()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 10
number of parameters: 3

id: 115
unit: def compute_rewards()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 3

id: 116
unit: def _get_chosen_prob()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 2

id: 117
unit: def __init__()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 27

id: 118
unit: def get_train_dataloader()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 1

id: 119
unit: def _create_model_from_path()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 12
number of parameters: 4

id: 120
unit: def __init__()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 23

id: 121
unit: def __init__()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 25

id: 122
unit: def __init__()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 16

id: 123
unit: def update()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 3
number of parameters: 2

id: 124
unit: def update()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 3

id: 125
unit: def flush_right()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 7
number of parameters: 2

id: 126
unit: def aesthetic_scorer()
file: trl/models/auxiliary_modules.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 2

id: 127
unit: def __init__()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 9
number of parameters: 3

id: 128
unit: def remove_hooks()
file: trl/models/utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 8
number of parameters: 1

id: 129
unit: def set_defaults_with_config()
file: trl/scripts/utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 8
number of parameters: 2

id: 130
unit: def init_communicator()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 1

id: 131
unit: def _compute_logprobs()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 4

id: 132
unit: def judge()
file: trl/trainer/judges.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 4

id: 133
unit: def __init__()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 26

id: 134
unit: def __init__()
file: trl/trainer/gkd_trainer.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 23

id: 135
unit: def flush_left()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 2

id: 136
unit: def __init__()
file: trl/models/auxiliary_modules.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 5

id: 137
unit: def unpack_tensor_single_stream()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 1

id: 138
unit: def show_text()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 139
unit: def show_colour_legend()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 1

id: 140
unit: def __init__()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 18

id: 141
unit: def __init__()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 20

id: 142
unit: def _shift_right()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 143
unit: def get_quantization_config()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 3
number of parameters: 1

id: 144
unit: def __init__()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 20

id: 145
unit: def _shift_right()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 2

id: 146
unit: def push_to_hub()
file: scripts/generate_tiny_models.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 4

id: 147
unit: def convert_state_dict()
file: trl/models/sd_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 6
number of parameters: 2

id: 148
unit: def get_trainable_layers()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 149
unit: def load_checkpoint()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 3

id: 150
unit: def get_git_commit_hash()
file: trl/scripts/utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 151
unit: def _pack_wrapped()
file: trl/data_utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 6
number of parameters: 2

id: 152
unit: def empty_device_cache()
file: trl/core.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 5
number of parameters: 1

id: 153
unit: def __call__()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 7
number of parameters: 4

id: 154
unit: def __init__()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 4

id: 155
unit: def _create_model_from_path()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 10
number of parameters: 3

id: 156
unit: def _tokenize()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 3

id: 157
unit: def generate_from_model()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 4

id: 158
unit: def generate_from_model()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 4

id: 159
unit: def add_hooks()
file: trl/models/utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 7
number of parameters: 1

id: 160
unit: def __init__()
file: trl/import_utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 6

id: 161
unit: def masked_var()
file: trl/core.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 162
unit: def append_segment()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 4

id: 163
unit: def _compute_rewards()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 4

id: 164
unit: def generate_on_policy_outputs()
file: trl/trainer/gkd_trainer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 4

id: 165
unit: def _compute_rewards()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 4

id: 166
unit: def _log_statistics()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 167
unit: def __init__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 168
unit: def __init__()
file: trl/trainer/rloo_trainer.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 19

id: 169
unit: def main()
file: scripts/add_copyrights.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 0

id: 170
unit: def init_zero_verbose()
file: trl/scripts/utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 0

id: 171
unit: def merge_models()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 172
unit: def maybe_convert_to_chatml()
file: trl/data_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 9
number of parameters: 2

id: 173
unit: def parse_tool_call()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 2

id: 174
unit: def _setup_optimizer()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 175
unit: def _enable_gradient_checkpointing()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 3

id: 176
unit: def _maybe_log_save_evaluate()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 9
number of parameters: 1

id: 177
unit: def _setup_optimizer()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 178
unit: def _enable_gradient_checkpointing()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 7
number of parameters: 3

id: 179
unit: def pad_to_length()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 5

id: 180
unit: def selective_log_softmax()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 2

id: 181
unit: def check_and_add_copyright()
file: scripts/add_copyrights.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 1

id: 182
unit: def save_pretrained()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 3

id: 183
unit: def __init__()
file: trl/models/auxiliary_modules.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 1

id: 184
unit: def generate()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 185
unit: def is_conversational()
file: trl/data_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 9
number of parameters: 2

id: 186
unit: def extract_prompt()
file: trl/data_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 187
unit: def __getattr__()
file: trl/import_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 188
unit: def flatten_dict()
file: trl/core.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 189
unit: def _vectorize_prompt()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 3

id: 190
unit: def _load_optimizer_and_scheduler()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 2

id: 191
unit: def _log_statistics()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 192
unit: def tokenize_row()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 7
number of parameters: 3

id: 193
unit: def __iter__()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 9
number of parameters: 1

id: 194
unit: def __init__()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 20

id: 195
unit: def print_rich_table()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 1

id: 196
unit: def generate_model_card()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 197
unit: def _get_variance()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 3

id: 198
unit: def create_merge_config_linear()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 1

id: 199
unit: def __init__()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 200
unit: def null_ref_context()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 1

id: 201
unit: def null_ref_context()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 1

id: 202
unit: def null_ref_context()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 1

id: 203
unit: def null_ref_context()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 1

id: 204
unit: def compute_ref_log_probs()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 205
unit: def remove_none_values()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 8
number of parameters: 1

id: 206
unit: def _get_train_sampler()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 11
number of parameters: 2

id: 207
unit: def sync_target_model()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 3

id: 208
unit: def __init__()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 209
unit: def on_train_end()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 5

id: 210
unit: def state_dict()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 3

id: 211
unit: def state_dict()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 3

id: 212
unit: def save_checkpoint()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 7
number of parameters: 4

id: 213
unit: def format_dataset()
file: trl/extras/dataset_formatting.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 1

id: 214
unit: def __init__()
file: trl/extras/best_of_n_sampler.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 215
unit: def profiling_context()
file: trl/extras/profiling.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 7
number of parameters: 2

id: 216
unit: def maybe_extract_prompt()
file: trl/data_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 12
number of parameters: 2

id: 217
unit: def save_model()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 3

id: 218
unit: def _forward()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 6

id: 219
unit: def _set_signature_columns_if_needed()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 1

id: 220
unit: def cross_entropy_loss()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 221
unit: def cross_entropy_loss()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 222
unit: def on_train_begin()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 5

id: 223
unit: def _merge_and_maybe_push()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 4

id: 224
unit: def get_tracked_python_files()
file: scripts/add_copyrights.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 0

id: 225
unit: def _get_current_device()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 226
unit: def _init_weights()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 2

id: 227
unit: def _init_weights()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 2

id: 228
unit: def scheduler_step()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 229
unit: def chat_template()
file: trl/models/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 1

id: 230
unit: def create()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 1

id: 231
unit: def close_communicator()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 232
unit: def _unpair_row()
file: trl/data_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 3

id: 233
unit: def pack_examples()
file: trl/data_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 3

id: 234
unit: def search()
file: trl/data_utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 2

id: 235
unit: def __post_init__()
file: trl/trainer/sft_config.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 236
unit: def bco_loss()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 237
unit: def _prepare_gradient_checkpointing()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 3

id: 238
unit: def _set_signature_columns_if_needed()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 1

id: 239
unit: def tokenize_row()
file: trl/trainer/prm_trainer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

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

id: 241
unit: def empty_cache()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 0

id: 242
unit: def _get_checkpoint_from_hub()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 243
unit: def __init__()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 6

id: 244
unit: def forward()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 245
unit: def __call__()
file: trl/models/auxiliary_modules.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 246
unit: def __init__()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 247
unit: def upload_model_to_hf()
file: trl/mergekit_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 248
unit: def update_named_param()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 249
unit: def add()
file: trl/data_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 250
unit: def remove()
file: trl/data_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 251
unit: def is_conversational_from_value()
file: trl/data_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 6
number of parameters: 2

id: 252
unit: def _get_module()
file: trl/import_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 253
unit: def generate()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 5
number of parameters: 2

id: 254
unit: def kto_loss()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 255
unit: def _prepare_dataset()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 256
unit: def log()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 6
number of parameters: 4

id: 257
unit: def __init__()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 258
unit: def _get_last_hidden_state()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 5

id: 259
unit: def compute_loss()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 5

id: 260
unit: def peft_module_casting_to_bf16()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 8
number of parameters: 1

id: 261
unit: def add_bos_token_if_needed()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 262
unit: def entropy_from_logits()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 263
unit: def __init__()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 264
unit: def forward()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 265
unit: def forward()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 266
unit: def _left_broadcast()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 267
unit: def get_act_offloading_ctx_manager()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 268
unit: def make_parser()
file: trl/scripts/dpo.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 269
unit: def make_parser()
file: trl/scripts/sft.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 270
unit: def make_parser()
file: trl/scripts/kto.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 271
unit: def init_communicator()
file: trl/scripts/vllm_serve.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 4

id: 272
unit: def update_named_param()
file: trl/scripts/vllm_serve.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 4

id: 273
unit: def make_parser()
file: trl/scripts/grpo.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 274
unit: def __init__()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 275
unit: def truncate()
file: trl/data_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 276
unit: def truncate()
file: trl/data_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 1

id: 277
unit: def tasks_end_check()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 3

id: 278
unit: def _save_checkpoint()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 279
unit: def __init__()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 280
unit: def _save_checkpoint()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 281
unit: def _save_optimizer_and_scheduler()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 5
number of parameters: 2

id: 282
unit: def evaluation_loop()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 283
unit: def _save_checkpoint()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 284
unit: def evaluation_loop()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 285
unit: def _save_checkpoint()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 286
unit: def _save_checkpoint()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 287
unit: def evaluation_loop()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 288
unit: def _save_checkpoint()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 289
unit: def _save_checkpoint()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 290
unit: def _compute_losses()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 291
unit: def _save_checkpoint()
file: trl/trainer/prm_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 292
unit: def step()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 293
unit: def _save_checkpoint()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 294
unit: def __init__()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 295
unit: def _save_checkpoint()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 296
unit: def prediction_step()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 297
unit: def _save_checkpoint()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 298
unit: def _save_checkpoint()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 299
unit: def evaluation_loop()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 300
unit: def _save_checkpoint()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 301
unit: def print_prompt_completions_sample()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 302
unit: def evaluation_loop()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 303
unit: def _save_checkpoint()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 304
unit: def _save_checkpoint()
file: trl/trainer/rloo_trainer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 305
unit: def _generate_completions()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 306
unit: def on_prediction_step()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 6

id: 307
unit: def forward()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 308
unit: def __init__()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 309
unit: def __init__()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 310
unit: def make_parser()
file: trl/scripts/vllm_serve.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 311
unit: def generate()
file: trl/extras/best_of_n_sampler.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 312
unit: def profiling_decorator()
file: trl/extras/profiling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 313
unit: def __dir__()
file: trl/import_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 1

id: 314
unit: def masked_whiten()
file: trl/core.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 315
unit: def randn_tensor()
file: trl/core.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 316
unit: def split_query_response_tokens()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 317
unit: def think_format_reward()
file: trl/rewards/format_rewards.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 3

id: 318
unit: def __post_init__()
file: trl/trainer/gkd_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 6
number of parameters: 1

id: 319
unit: def train()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 320
unit: def get_batch_logps()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 321
unit: def _get_udm_weight()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 322
unit: def compute_loss()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 323
unit: def _get_train_sampler()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 324
unit: def prediction_step()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 325
unit: def mixture_coef()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 326
unit: def judge()
file: trl/trainer/judges.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 327
unit: def get_batch_logps()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 328
unit: def compute_loss()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 329
unit: def _get_train_sampler()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 330
unit: def prediction_step()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 331
unit: def beta()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 332
unit: def _prepare_dataset()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 333
unit: def dpo_loss()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 334
unit: def compute_loss()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 335
unit: def prediction_step()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 336
unit: def log()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 4

id: 337
unit: def _prepare_model_for_kbit_training()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 338
unit: def add_eos()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 2

id: 339
unit: def alpha()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 340
unit: def _step_safety_checker()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 341
unit: def collator()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 1

id: 342
unit: def train()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 343
unit: def _get_eval_sampler()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 344
unit: def compute_loss()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 345
unit: def prediction_step()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 346
unit: def concatenated_inputs()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 347
unit: def get_batch_logps()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 348
unit: def compute_loss()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 349
unit: def prediction_step()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 350
unit: def log()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 4

id: 351
unit: def save_to_json()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 2

id: 352
unit: def truncate_response()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 353
unit: def batch_generation()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 354
unit: def get_comet_experiment_url()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 0

id: 355
unit: def log_table_to_comet_experiment()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 356
unit: def concatenated_inputs()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 357
unit: def get_batch_logps()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 358
unit: def compute_loss()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 359
unit: def prediction_step()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 360
unit: def log()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 4

id: 361
unit: def __post_init__()
file: trl/trainer/ddpo_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 362
unit: def on_step_end()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 5

id: 363
unit: def __init__()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 364
unit: def __post_init__()
file: trl/trainer/alignprop_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 365
unit: def prepare_inputs_for_generation()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 366
unit: def _has_lm_head()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 1

id: 367
unit: def save_pretrained()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 368
unit: def setup_chat_format()
file: trl/models/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 369
unit: def clone_chat_template()
file: trl/models/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 370
unit: def wrapped_forward()
file: trl/models/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 371
unit: def close_communicator()
file: trl/scripts/vllm_serve.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 372
unit: def parse_args_and_config()
file: trl/scripts/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 373
unit: def reset_prefix_cache()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 374
unit: def masked_mean()
file: trl/core.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 375
unit: def compute_reward()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 3

id: 376
unit: def _generate_batched()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 377
unit: def __post_init__()
file: trl/trainer/model_config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 5
number of parameters: 1

id: 378
unit: def __init__()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 379
unit: def create_model_card()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 380
unit: def loss()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 381
unit: def create_model_card()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 382
unit: def get_batch_loss_metrics()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 383
unit: def create_model_card()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 384
unit: def _compute_losses()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 385
unit: def create_model_card()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 386
unit: def create_model_card()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 387
unit: def create_model_card()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 388
unit: def get_batch_loss_metrics()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 389
unit: def create_model_card()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 390
unit: def create_model_card()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 391
unit: def create_model_card()
file: trl/trainer/gkd_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 392
unit: def create_model_card()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 393
unit: def __post_init__()
file: trl/trainer/online_dpo_config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 1

id: 394
unit: def create_model_card()
file: trl/trainer/prm_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 395
unit: def create_model_card()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 396
unit: def __post_init__()
file: trl/trainer/iterative_sft_config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 397
unit: def compute_rewards()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 398
unit: def loss()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 399
unit: def create_model_card()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 400
unit: def nanstd()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 401
unit: def shuffle_tensor_dict()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 5
number of parameters: 2

id: 402
unit: def create_model_card()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 403
unit: def create_model_card()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 404
unit: def get_batch_loss_metrics()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 405
unit: def create_model_card()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 406
unit: def __post_init__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 407
unit: def pad()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 408
unit: def exact_div()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 409
unit: def get_kbit_device_map()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 0

id: 410
unit: def get_batch_loss_metrics()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 411
unit: def create_model_card()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 412
unit: def to_dict()
file: trl/trainer/ddpo_config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 413
unit: def create_model_card()
file: trl/trainer/rloo_trainer.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 414
unit: def on_evaluate()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 5

id: 415
unit: def on_predict()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 5

id: 416
unit: def __init__()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 417
unit: def to_dict()
file: trl/trainer/alignprop_config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 418
unit: def unwrap_model_for_generation()
file: trl/models/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 419
unit: def verify_sufficient_virtual_memory()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 420
unit: def __init__()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

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

id: 422
unit: def apply_chat_template()
file: trl/data_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 423
unit: def maybe_apply_chat_template()
file: trl/data_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

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

id: 425
unit: def __post_init__()
file: trl/trainer/nash_md_config.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 426
unit: def __post_init__()
file: trl/trainer/xpo_config.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 427
unit: def forward()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 428
unit: def _tokenize()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 429
unit: def _get_kl_dataset()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 430
unit: def get_batch_loss_metrics()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 431
unit: def shift_tokens_right()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 432
unit: def nanmin()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 433
unit: def nanmax()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 434
unit: def evaluate()
file: trl/trainer/reward_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 435
unit: def cpo_loss()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 436
unit: def load_from_json()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 437
unit: def disable_dropout_in_model()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 438
unit: def __init__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 439
unit: def get_exp_cap()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 440
unit: def first_true_indices()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 441
unit: def forward()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 442
unit: def odds_ratio_loss()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 443
unit: def __init__()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 444
unit: def on_step_end()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 5

id: 445
unit: def on_log()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 6

id: 446
unit: def push_to_hub()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 447
unit: def state_dict()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 448
unit: def post_init()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 449
unit: def generate()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 450
unit: def push_to_hub()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 451
unit: def push_to_hub()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 452
unit: def generate()
file: trl/models/modeling_value_head.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 453
unit: def get_tensor_id()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 0

id: 454
unit: def chunk_list()
file: trl/scripts/vllm_serve.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 455
unit: def update_model_params()
file: trl/extras/vllm_client.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

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

id: 457
unit: def complete()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 458
unit: def last_text_segment()
file: trl/environment/base_environment.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 459
unit: def __post_init__()
file: trl/trainer/dpo_config.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 460
unit: def __post_init__()
file: trl/trainer/cpo_config.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 461
unit: def _save_model_hook()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 462
unit: def _load_model_hook()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 463
unit: def _save_pretrained()
file: trl/trainer/ddpo_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 464
unit: def __post_init__()
file: trl/trainer/prm_config.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 465
unit: def __post_init__()
file: trl/trainer/bco_config.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 466
unit: def store_metrics()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 5

id: 467
unit: def __post_init__()
file: trl/trainer/reward_config.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 468
unit: def _tokenize()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 469
unit: def store_metrics()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 5

id: 470
unit: def store_metrics()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 5

id: 471
unit: def training_step()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 472
unit: def __post_init__()
file: trl/trainer/kto_config.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 473
unit: def _create_model_from_path()
file: trl/trainer/iterative_sft_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 474
unit: def calculate_loss()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 475
unit: def _save_model_hook()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 476
unit: def _load_model_hook()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 477
unit: def _save_pretrained()
file: trl/trainer/alignprop_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 478
unit: def _set_signature_columns_if_needed()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 479
unit: def store_metrics()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 5

id: 480
unit: def __post_init__()
file: trl/trainer/orpo_config.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 481
unit: def cap_exp()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 482
unit: def __post_init__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 483
unit: def decode_and_strip_padding()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 484
unit: def store_metrics()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 5

id: 485
unit: def _sync_target_model()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 486
unit: def on_save()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 6

id: 487
unit: def on_train_end()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 6

id: 488
unit: def __init__()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 489
unit: def add_and_load_reward_modeling_adapter()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 490
unit: def create_reference_model()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 491
unit: def __call__()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 492
unit: def _validate_model_class()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 493
unit: def _validate_model_kwargs()
file: trl/models/modeling_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 494
unit: def forward()
file: trl/models/auxiliary_modules.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 495
unit: def __call__()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 496
unit: def scheduler_step()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 497
unit: def unet()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 498
unit: def vae()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 499
unit: def tokenizer()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 500
unit: def scheduler()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 501
unit: def text_encoder()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 502
unit: def autocast()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 503
unit: def set_progress_bar_config()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 504
unit: def save_pretrained()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 505
unit: def get_trainable_layers()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 506
unit: def save_checkpoint()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 507
unit: def load_checkpoint()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 508
unit: def __call__()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 509
unit: def rgb_with_grad()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 510
unit: def scheduler_step()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 511
unit: def unet()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 512
unit: def vae()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 513
unit: def tokenizer()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 514
unit: def scheduler()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 515
unit: def text_encoder()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 516
unit: def autocast()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 517
unit: def set_progress_bar_config()
file: trl/models/modeling_sd_base.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 518
unit: def system()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 519
unit: def user()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 520
unit: def assistant()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 521
unit: def get_all_parameters()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 522
unit: def iter_params()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 523
unit: def __call__()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 524
unit: def on_after_inner_forward()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 525
unit: def on_after_outer_forward()
file: trl/models/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 526
unit: def get_cpu_ram_pct()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 527
unit: def get_num_bytes_tensor()
file: trl/models/activation_offloading.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 528
unit: def llm_worker()
file: trl/scripts/vllm_serve.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 529
unit: def conversations_formatting_function()
file: trl/extras/dataset_formatting.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 530
unit: def get_formatting_func_from_dataset()
file: trl/extras/dataset_formatting.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 531
unit: def unpair_preference_dataset()
file: trl/data_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 532
unit: def maybe_unpair_preference_dataset()
file: trl/data_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 533
unit: def pack_dataset()
file: trl/data_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 534
unit: def truncate_dataset()
file: trl/data_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 535
unit: def is_deepspeed_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 536
unit: def is_diffusers_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 537
unit: def is_fastapi_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 538
unit: def is_liger_kernel_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 539
unit: def is_llm_blender_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 540
unit: def is_mergekit_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 541
unit: def is_pydantic_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 542
unit: def is_requests_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 543
unit: def is_unsloth_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 544
unit: def is_uvicorn_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 545
unit: def is_vllm_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 546
unit: def is_vllm_ascend_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 547
unit: def is_joblib_available()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 548
unit: def __reduce__()
file: trl/import_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

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

id: 550
unit: def __call__()
file: trl/core.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 551
unit: def get_train_dataloader()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 552
unit: def get_eval_dataloader()
file: trl/trainer/ppo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 553
unit: def make_inputs_require_grad()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 554
unit: def make_inputs_require_grad()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 555
unit: def match_underlying_distribution()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 556
unit: def _get_prompt_embeddings()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 557
unit: def forward()
file: trl/trainer/bco_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 558
unit: def gather_mean()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 559
unit: def training_step()
file: trl/trainer/nash_md_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 560
unit: def judge()
file: trl/trainer/judges.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 561
unit: def judge()
file: trl/trainer/judges.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 562
unit: def make_inputs_require_grad()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 563
unit: def make_inputs_require_grad()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 564
unit: def forward()
file: trl/trainer/kto_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 565
unit: def training_step()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 566
unit: def _maybe_log_save_evaluate()
file: trl/trainer/online_dpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 567
unit: def _prepare_peft_model()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 568
unit: def concatenated_inputs()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 569
unit: def concatenated_forward()
file: trl/trainer/dpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 570
unit: def _func()
file: trl/trainer/sft_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 571
unit: def generalized_jsd_loss()
file: trl/trainer/gkd_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 572
unit: def training_step()
file: trl/trainer/gkd_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 573
unit: def gather_mean()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 574
unit: def training_step()
file: trl/trainer/xpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 575
unit: def __len__()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 576
unit: def split_tensor_dict()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 577
unit: def identity()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 578
unit: def _get_per_token_logps_and_entropies()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 579
unit: def _prepare_inputs()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 580
unit: def _generate_and_score_completions()
file: trl/trainer/grpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 581
unit: def make_inputs_require_grad()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 582
unit: def make_inputs_require_grad()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 583
unit: def concatenated_forward()
file: trl/trainer/cpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 584
unit: def __len__()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 585
unit: def get_global_statistics()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 586
unit: def get_stats()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 587
unit: def get_reward()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 588
unit: def prepare_deepspeed()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 589
unit: def generate()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 590
unit: def add_eos_token_if_needed()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 591
unit: def truncate_right()
file: trl/trainer/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 592
unit: def make_inputs_require_grad()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 593
unit: def make_inputs_require_grad()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 594
unit: def concatenated_forward()
file: trl/trainer/orpo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 595
unit: def get_train_dataloader()
file: trl/trainer/rloo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 596
unit: def get_eval_dataloader()
file: trl/trainer/rloo_trainer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 597
unit: def _win_rate_completions_df()
file: trl/trainer/callbacks.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0