def get_model_validator()

in src/hyperpod_nemo_adapter/conf/config_schemas.py [0:0]

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

def get_model_validator(use_smp_model, extra="forbid") -> type[BaseModel]:
    global smp
    smp = use_smp_model
    if extra == "forbid":
        if use_smp_model:
            return ConfigWithSMPForbid
        else:
            return ConfigForbid
    elif extra == "allow":
        if use_smp_model:
            return ConfigWithSMPAllow
        else:
            return ConfigAllow
    else:
        raise ValueError(f"Unsupported extra type {extra}")