id: 1
unit: def main()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 851 LOC
McCabe index: 91
number of parameters: 0

id: 2
unit: def main()
file: training/run_distillation.py
start line: 0
end line: 0
size: 772 LOC
McCabe index: 103
number of parameters: 0

id: 3
unit: def main()
file: training/run_pseudo_labelling.py
start line: 0
end line: 0
size: 460 LOC
McCabe index: 56
number of parameters: 0

id: 4
unit: def main()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 458 LOC
McCabe index: 95
number of parameters: 0

id: 5
unit: def main()
file: training/run_eval.py
start line: 0
end line: 0
size: 380 LOC
McCabe index: 81
number of parameters: 0

id: 6
unit: def main()
file: training/flax/run_pseudo_labelling_pt.py
start line: 0
end line: 0
size: 366 LOC
McCabe index: 56
number of parameters: 0

id: 7
unit: def main()
file: training/flax/run_speed_pt.py
start line: 0
end line: 0
size: 330 LOC
McCabe index: 54
number of parameters: 0

id: 8
unit: def main()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 316 LOC
McCabe index: 88
number of parameters: 0

id: 9
unit: def main()
file: training/flax/run_long_form_transcription.py
start line: 0
end line: 0
size: 214 LOC
McCabe index: 67
number of parameters: 0

id: 10
unit: def main()
file: training/flax/run_pt_long_form_transcription.py
start line: 0
end line: 0
size: 189 LOC
McCabe index: 58
number of parameters: 0

id: 11
unit: def __call__()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 128 LOC
McCabe index: 23
number of parameters: 2

id: 12
unit: def main()
file: training/flax/convert_train_state_to_hf.py
start line: 0
end line: 0
size: 110 LOC
McCabe index: 25
number of parameters: 0

id: 13
unit: def preprocess_batch()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 74 LOC
McCabe index: 13
number of parameters: 5

id: 14
unit: def parse_args()
file: training/flax/create_student_model.py
start line: 0
end line: 0
size: 58 LOC
McCabe index: 1
number of parameters: 0

id: 15
unit: def parse_args()
file: training/create_student_model.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 1
number of parameters: 0

id: 16
unit: def get_forced_decoder_ids()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 39 LOC
McCabe index: 17
number of parameters: 5

id: 17
unit: def _concatenate_to_cache()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 16
number of parameters: 5

id: 18
unit: def setup()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 1
number of parameters: 1

id: 19
unit: def setup()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 1
number of parameters: 1

id: 20
unit: def init_cache()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 1
number of parameters: 4

id: 21
unit: def __call__()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 4
number of parameters: 2

id: 22
unit: def setup()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 3
number of parameters: 1

id: 23
unit: def __call__()
file: training/flax/run_pseudo_labelling_pt.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 9
number of parameters: 4

id: 24
unit: def __call__()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 8
number of parameters: 4

id: 25
unit: def __call__()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 10
number of parameters: 4

id: 26
unit: def setup()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 1
number of parameters: 1

id: 27
unit: def __call__()
file: training/run_distillation.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 6
number of parameters: 4

id: 28
unit: def get_data_layout()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 6
number of parameters: 3

id: 29
unit: def __call__()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 6
number of parameters: 4

id: 30
unit: def __call__()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 2
number of parameters: 4

id: 31
unit: def init_weights()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 3
number of parameters: 4

id: 32
unit: def variance_scaling()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 7
number of parameters: 6

id: 33
unit: def __call__()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 8
number of parameters: 4

id: 34
unit: def chunk_iter_with_batch()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 5
number of parameters: 6

id: 35
unit: def __post_init__()
file: training/run_pseudo_labelling.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 8
number of parameters: 1

id: 36
unit: def setup()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 1
number of parameters: 1

id: 37
unit: def convert_unroll_to_scan()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 10
number of parameters: 3

id: 38
unit: def __call__()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 3
number of parameters: 2

id: 39
unit: def canonicalize_padding()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 3
number of parameters: 2

id: 40
unit: def language_to_id()
file: training/run_eval.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 5
number of parameters: 2

id: 41
unit: def __call__()
file: training/run_pseudo_labelling.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 8
number of parameters: 4

id: 42
unit: def get_gpu_mesh()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 1

id: 43
unit: def convert_scan_to_unroll()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 15
number of parameters: 3

id: 44
unit: def _relative_position_bucket()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 4

id: 45
unit: def postprocess()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 4

id: 46
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 4

id: 47
unit: def __init__()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 2

id: 48
unit: def create()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 2

id: 49
unit: def setup()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 1

id: 50
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 51
unit: def setup()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 1

id: 52
unit: def __call__()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 53
unit: def get_mesh_axes()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 2

id: 54
unit: def enable_scan()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 1

id: 55
unit: def disable_scan()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 1

id: 56
unit: def decode()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 57
unit: def decode()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 0

id: 58
unit: def load_multiple_datasets()
file: training/run_distillation.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 59
unit: def get_parameter_names()
file: training/run_distillation.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 3

id: 60
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 61
unit: def _compute_fans()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 3

id: 62
unit: def __call__()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 2

id: 63
unit: def sorted_checkpoints()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 8
number of parameters: 3

id: 64
unit: def apply_gradients()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 5

id: 65
unit: def forward()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 66
unit: def apply_gradients()
file: training/flax/convert_train_state_to_hf.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 4

id: 67
unit: def sorted_checkpoints()
file: training/run_distillation.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 9
number of parameters: 2

id: 68
unit: def rotate_checkpoints()
file: training/run_distillation.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 4

id: 69
unit: def _np_extract_fbank_features()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 70
unit: def generate()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 71
unit: def move_params_to_devices()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 3

id: 72
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 73
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 74
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 75
unit: def __init__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 0

id: 76
unit: def encode()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 77
unit: def combine_masks()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 2

id: 78
unit: def combine_biases()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 6
number of parameters: 1

id: 79
unit: def _np_extract_fbank_features()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 80
unit: def get_data_loader()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 81
unit: def save_state()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 4

id: 82
unit: def load_multiple_datasets()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 83
unit: def _np_extract_fbank_features()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 84
unit: def sorted_best_checkpoints()
file: training/run_distillation.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 7
number of parameters: 2

id: 85
unit: def get_last_checkpoint()
file: training/run_distillation.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 5
number of parameters: 1

id: 86
unit: def generate()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 87
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 88
unit: def generate()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 89
unit: def dot_product_attention()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 90
unit: def write_metric()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 6

id: 91
unit: def create()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 7

id: 92
unit: def log_pred()
file: training/run_distillation.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 93
unit: def __call__()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 94
unit: def _convert_to_activation_function()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 95
unit: def write_wandb_pred()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 96
unit: def get_data_loader()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 97
unit: def save_hf_weights()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 98
unit: def init_student_model_from_teacher()
file: training/flax/create_student_model.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 99
unit: def init_student_model_from_teacher()
file: training/create_student_model.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 100
unit: def write_wandb_pred()
file: training/run_eval.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 101
unit: def __post_init__()
file: training/run_pseudo_labelling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 102
unit: def log_pred()
file: training/run_pseudo_labelling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 103
unit: def __post_init__()
file: training/run_distillation.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 104
unit: def log_metric()
file: training/run_distillation.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 105
unit: def dh_dd_mh_md()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 106
unit: def __init__()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 107
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 108
unit: def enable_gradient_checkpointing()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 109
unit: def prepare_inputs_for_generation()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 110
unit: def state_dict()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 111
unit: def restore_state()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 112
unit: def setup()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 113
unit: def log_pred()
file: training/flax/run_pseudo_labelling_pt.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 114
unit: def write_wandb_pred()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 5

id: 115
unit: def write_train_metric()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 5

id: 116
unit: def write_wandb_pred()
file: training/flax/run_long_form_transcription.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 117
unit: def write_wandb_pred()
file: training/flax/run_pt_long_form_transcription.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 118
unit: def convert_dataset_str_to_list()
file: training/run_distillation.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 119
unit: def __init__()
file: training/flax/distil_whisper/pipeline.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 120
unit: def pjit()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 121
unit: def pjit_with_cpu_fallback()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 122
unit: def partition()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 123
unit: def partition()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 124
unit: def partition()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 125
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 126
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 127
unit: def __call__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 128
unit: def pipeline_generate()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 129
unit: def _validate_params_axes()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 130
unit: def as_logical_axes()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 131
unit: def write_wandb_metric()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 6

id: 132
unit: def convert_dataset_str_to_list()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 133
unit: def get_layers_to_supervise()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 134
unit: def bounds_from_last_device()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 135
unit: def get_mesh()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 136
unit: def __init__()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 137
unit: def __init__()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 138
unit: def _decoder_forward()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 139
unit: def _decoder_forward()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 140
unit: def make_attention_mask()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 141
unit: def _conv_dimension_numbers()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 142
unit: def get_data_loader()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 143
unit: def write_wandb_metric()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 144
unit: def write_wandb_metric()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 5

id: 145
unit: def write_wandb_metric()
file: training/flax/run_long_form_transcription.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 146
unit: def write_wandb_metric()
file: training/flax/run_speed_pt.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 147
unit: def write_wandb_metric()
file: training/flax/run_pt_long_form_transcription.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 148
unit: def write_wandb_metric()
file: training/run_eval.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 149
unit: def shift_tokens_right()
file: training/run_pseudo_labelling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 150
unit: def log_metric()
file: training/run_pseudo_labelling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 151
unit: def with_sharding_constraint()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

id: 152
unit: def get_cpu_mesh()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 153
unit: def __init__()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 4

id: 154
unit: def nd_dense_init()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 155
unit: def _canonicalize_tuple()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 156
unit: def make_decoder_mask()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 157
unit: def shift_tokens_right()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 158
unit: def shift_tokens_right()
file: training/flax/run_pseudo_labelling_pt.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 159
unit: def log_metric()
file: training/flax/run_pseudo_labelling_pt.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 160
unit: def shift_tokens_right()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 161
unit: def shift_tokens_right()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 162
unit: def get_coords()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 163
unit: def default_mesh()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 164
unit: def standard_logical_axis_rules()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 165
unit: def update_inputs_for_generation()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 166
unit: def write_metric()
file: training/run_eval.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 4

id: 167
unit: def global_mesh_defined()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 168
unit: def _id_fn()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 169
unit: def __call__()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 170
unit: def lower()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 171
unit: def _encoder_forward()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 172
unit: def attend()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 173
unit: def make_causal_mask()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 174
unit: def write_metric()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 4

id: 175
unit: def write_eval_metric()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 4

id: 176
unit: def write_metric()
file: training/flax/run_long_form_transcription.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 177
unit: def write_metric()
file: training/flax/run_speed_pt.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 4

id: 178
unit: def write_metric()
file: training/flax/run_pt_long_form_transcription.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 3

id: 179
unit: def data()
file: training/flax/run_pt_long_form_transcription.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 3
number of parameters: 3

id: 180
unit: def convert_dataset_str_to_list()
file: training/run_eval.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 181
unit: def __new__()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 182
unit: def __repr__()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 183
unit: def get_local_chunk_info()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 184
unit: def mesh()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 185
unit: def data_partition_spec()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 186
unit: def get_local_chunk_info()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 187
unit: def params_on_devices()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 188
unit: def _local_chunker()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 189
unit: def get_logical_axes()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 190
unit: def get_mesh_axes()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 191
unit: def compile()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 192
unit: def _local_chunker()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 193
unit: def mesh()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 194
unit: def compile()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 195
unit: def logical_axis_rules()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 196
unit: def get_logical_axes()
file: training/flax/distil_whisper/partitioner.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 197
unit: def _split_heads()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 198
unit: def _merge_heads()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 199
unit: def _get_encoder_module()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 200
unit: def _get_decoder_module()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 201
unit: def _get_encoder_module()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 202
unit: def _get_decoder_module()
file: training/flax/distil_whisper/modeling_flax_whisper.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 203
unit: def _split_variables_and_axes()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 204
unit: def param_states()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 205
unit: def apply_gradient()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 206
unit: def replace_step()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 207
unit: def replace_params()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 208
unit: def replace_flax_mutables()
file: training/flax/distil_whisper/train_state.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 209
unit: def query_init()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 210
unit: def swap_dims()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 211
unit: def _normalize_axes()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 2

id: 212
unit: def shared_weights()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 213
unit: def shared_weights()
file: training/flax/distil_whisper/layers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 214
unit: def convert_audio_to_wandb()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 215
unit: def convert_dataset_str_to_list()
file: training/flax/run_eval.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 216
unit: def replicate()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 217
unit: def create_learning_rate_fn()
file: training/flax/run_finetuning.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 218
unit: def rotate_checkpoints()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 219
unit: def to_fp32()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 220
unit: def to_bf16()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 221
unit: def replicate()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 222
unit: def unreplicate()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 223
unit: def write_wandb_pred()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 224
unit: def create_learning_rate_fn()
file: training/flax/run_distillation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 225
unit: def convert_audio_to_wandb()
file: training/flax/run_long_form_transcription.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 226
unit: def convert_dataset_str_to_list()
file: training/flax/run_long_form_transcription.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 227
unit: def create_learning_rate_fn()
file: training/flax/convert_train_state_to_hf.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 228
unit: def replicate()
file: training/flax/convert_train_state_to_hf.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 229
unit: def unreplicate()
file: training/flax/convert_train_state_to_hf.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 230
unit: def convert_dataset_str_to_list()
file: training/flax/run_speed_pt.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 231
unit: def convert_audio_to_wandb()
file: training/flax/run_pt_long_form_transcription.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 232
unit: def convert_dataset_str_to_list()
file: training/flax/run_pt_long_form_transcription.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0