Spaces:

SyncShift
/

sql-correction-env

Sleeping

sql-correction-env / sql_env /tasks /__init__.py

Fix server.py, expand tasks to 40, add tests, clean observation schema

2194233 7 months ago

354 Bytes

from sql_env.tasks.easy import EASY_TASKS
from sql_env.tasks.medium import MEDIUM_TASKS
from sql_env.tasks.hard import HARD_TASKS

ALL_TASKS = {
    "easy": EASY_TASKS,
    "medium": MEDIUM_TASKS,
    "hard": HARD_TASKS,
}

# Alias used by server.py
TASK_SETS = ALL_TASKS

__all__ = ["ALL_TASKS", "TASK_SETS", "EASY_TASKS", "MEDIUM_TASKS", "HARD_TASKS"]

	from sql_env.tasks.easy import EASY_TASKS
	from sql_env.tasks.medium import MEDIUM_TASKS
	from sql_env.tasks.hard import HARD_TASKS

	ALL_TASKS = {
	"easy": EASY_TASKS,
	"medium": MEDIUM_TASKS,
	"hard": HARD_TASKS,
	}

	# Alias used by server.py
	TASK_SETS = ALL_TASKS

	__all__ = ["ALL_TASKS", "TASK_SETS", "EASY_TASKS", "MEDIUM_TASKS", "HARD_TASKS"]