Function get_runner_class

src/utils.py:17–22

(model_id)

def get_runner_class(model_id):
    cls = id2worker_class.get(model_id, None)
    if cls is None:
        print(f"{model_id} is not registered in src/constants.py")
        exit(1)
    return cls


def match_choice(text, cot_flag):