Project-I/models/model_selector.py

from models.STDEN.stden_model import STDENModel

def model_selector(config):
    model_name = config['basic']['model']
    model = None
    match model_name:
        case 'STDEN': model = STDENModel(config)
    return model