id: 1
unit: def from_pretrained()
file: src/sal/models/skywork_o1_prm/modeling_base.py
start line: 0
end line: 0
size: 146 LOC
McCabe index: 32
number of parameters: 4

id: 2
unit: def _beam_search()
file: src/sal/search/beam_search.py
start line: 0
end line: 0
size: 137 LOC
McCabe index: 44
number of parameters: 4

id: 3
unit: def _dvts()
file: src/sal/search/diverse_verifier_tree_search.py
start line: 0
end line: 0
size: 107 LOC
McCabe index: 24
number of parameters: 4

id: 4
unit: def strip_string()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 81 LOC
McCabe index: 28
number of parameters: 2

id: 5
unit: def extract_answer()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 59 LOC
McCabe index: 25
number of parameters: 3

id: 6
unit: def best_of_n()
file: src/sal/search/best_of_n.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 22
number of parameters: 4

id: 7
unit: def symbolic_equal()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 12
number of parameters: 2

id: 8
unit: def score()
file: src/sal/utils/score.py
start line: 0
end line: 0
size: 40 LOC
McCabe index: 5
number of parameters: 2

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

id: 10
unit: def save_dataset()
file: src/sal/utils/data.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 6
number of parameters: 2

id: 11
unit: def _score_single()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 34 LOC
McCabe index: 6
number of parameters: 3

id: 12
unit: def _fix_fracs()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 8
number of parameters: 1

id: 13
unit: def choice_answer_clean()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 10
number of parameters: 1

id: 14
unit: def post_init()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 9
number of parameters: 2

id: 15
unit: def dvts()
file: src/sal/search/diverse_verifier_tree_search.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 8
number of parameters: 4

id: 16
unit: def memoized_canonical_form()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 6
number of parameters: 2

id: 17
unit: def find_box()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 1

id: 18
unit: def prepare_input()
file: src/sal/models/skywork_o1_prm/io_utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 3
number of parameters: 4

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

id: 20
unit: def __init__()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 9
number of parameters: 3

id: 21
unit: def beam_search()
file: src/sal/search/beam_search.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 8
number of parameters: 4

id: 22
unit: def prepare_batch_input_for_model()
file: src/sal/models/skywork_o1_prm/io_utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 3

id: 23
unit: def clean_units()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 5
number of parameters: 1

id: 24
unit: def _split_kwargs()
file: src/sal/models/skywork_o1_prm/modeling_base.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 2

id: 25
unit: def _fix_a_slash_b()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 1

id: 26
unit: def parse_digits()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 3
number of parameters: 1

id: 27
unit: def find_answer_with_largest_sum()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 7
number of parameters: 2

id: 28
unit: def find_majority_answer()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 9
number of parameters: 1

id: 29
unit: def make_step_rewards()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 30
unit: def __init__()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 3

id: 31
unit: def load_prm()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 1

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

id: 33
unit: def state_dict()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 3

id: 34
unit: def call_with_timeout()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 4

id: 35
unit: def subsample_completions()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 3

id: 36
unit: def load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 1

id: 37
unit: def choice_answer_clean()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 1

id: 38
unit: def derive_step_rewards()
file: src/sal/models/skywork_o1_prm/io_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 2

id: 39
unit: def load_single_revision()
file: scripts/merge_chunks.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 1

id: 40
unit: def str_to_pmatrix()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 41
unit: def compute_pass_at_k()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 42
unit: def _init_weights()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 2

id: 43
unit: def extract_multi_choice_answer()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 44
unit: def compute_weighted_pred()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 45
unit: def is_transformers_greater_than()
file: src/sal/models/skywork_o1_prm/modeling_base.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

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

id: 47
unit: def forward()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 48
unit: def get_dataset()
file: src/sal/utils/data.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 49
unit: def compute_naive_pred()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 50
unit: def __init__()
file: src/sal/models/skywork_o1_prm/modeling_base.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 51
unit: def convert_word_number()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 52
unit: def math_equal()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 53
unit: def score()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 54
unit: def add_and_load_reward_modeling_adapter()
file: src/sal/models/skywork_o1_prm/modeling_base.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 55
unit: def forward()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 56
unit: def generate_k_steps()
file: src/sal/search/utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 57
unit: def get_dataset_revisions()
file: src/sal/utils/hub.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 4
number of parameters: 1

id: 58
unit: def batched_math_shepherd_inference()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 59
unit: def last()
file: src/sal/search/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 60
unit: def list_mean()
file: src/sal/search/utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 61
unit: def pass_at_k()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 62
unit: def _get_current_device()
file: src/sal/models/skywork_o1_prm/modeling_base.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 63
unit: def create_reference_model()
file: src/sal/models/skywork_o1_prm/modeling_base.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 64
unit: def _fix_sqrt()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 65
unit: def symbolic_equal_process()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 66
unit: def compute_maj_pred()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 67
unit: def __init__()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

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

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

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

id: 71
unit: def generate()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 72
unit: def push_to_hub()
file: src/sal/models/skywork_o1_prm/prm_model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 73
unit: def is_digit()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 74
unit: def numeric_equal()
file: src/sal/utils/qwen_math_parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 75
unit: def parse_yaml_and_args()
file: src/sal/utils/parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 76
unit: def parse()
file: src/sal/utils/parser.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 77
unit: def aggregate_scores()
file: src/sal/utils/score.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 78
unit: def timeout_handler()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 79
unit: def extract_completion_answers()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 80
unit: def compute_level()
file: src/sal/utils/math.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 81
unit: def load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 82
unit: def score()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 83
unit: def score()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 84
unit: def load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 85
unit: def _score_batched()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 86
unit: def _load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 87
unit: def score()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 88
unit: def _load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 89
unit: def score()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 90
unit: def load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 91
unit: def load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 92
unit: def load_model_and_tokenizer()
file: src/sal/models/reward_models.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 93
unit: def build_conv()
file: src/sal/search/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0