def create_trainer_for_config()

in src/jobs/TuneGenTopicModel.py [0:0]

• 11 lines of code
• 5 McCabe index (conditional complexity)

def create_trainer_for_config(config: dict[str, any]):
    if "t5" in config["model_name"]:
        if config.get("teacher_model_artifact", None) is None:
            return TuneTopicT5(**config)
        else:
            return DistillTopicT5(**config)
    if "gpt" in config["model_name"]:
        return TuneTopicGPT2(**config)
    if "bart" in config["model_name"]:
        return TuneTopicBart(**config)
    return None