mirror of
https://github.com/princeton-nlp/tree-of-thought-llm
synced 2024-11-25 05:19:30 +00:00
12 lines
382 B
Python
12 lines
382 B
Python
|
def get_task(name, file=None):
|
||
|
|
if name == 'game24':
|
||
|
|
from .game24 import Game24Task
|
||
|
|
return Game24Task(file)
|
||
|
|
elif name == 'text':
|
||
|
|
from .text import TextTask
|
||
|
|
return TextTask(file)
|
||
|
|
elif name == 'crosswords':
|
||
|
|
from .crosswords import MiniCrosswordsTask
|
||
|
|
return MiniCrosswordsTask(file)
|
||
|
|
else:
|
||
|
|
raise NotImplementedError
|