def METHOD_NAME(name: str) -> Type[BaseModel]:
    for cls in get_all_subclasses(BaseModel):
        if cls.__name__ == name:
            return cls
    raise ValueError(f"Could not find model with name {name}")