def get_callbacks()

in src/open_r1/utils/callbacks.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def get_callbacks(train_config, model_config) -> List[TrainerCallback]:
    callbacks = []
    for callback_name in train_config.callbacks:
        if callback_name not in CALLBACKS:
            raise ValueError(f"Callback {callback_name} not found in CALLBACKS.")
        callbacks.append(CALLBACKS[callback_name](model_config))

    return callbacks