Spaces:

SyncShift
/

sql-correction-env

Sleeping

sql-correction-env / sql_env /__init__.py

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

2194233 7 months ago

335 Bytes

from sql_env.env import SQLCorrectionEnv
from sql_env.models import SQLObservation, SQLAction, SQLReward, SQLTask, StepResult
from sql_env.tasks import ALL_TASKS, TASK_SETS

__all__ = [
    "SQLCorrectionEnv",
    "SQLObservation",
    "SQLAction",
    "SQLReward",
    "SQLTask",
    "StepResult",
    "ALL_TASKS",
    "TASK_SETS",
]

	from sql_env.env import SQLCorrectionEnv
	from sql_env.models import SQLObservation, SQLAction, SQLReward, SQLTask, StepResult
	from sql_env.tasks import ALL_TASKS, TASK_SETS

	__all__ = [
	"SQLCorrectionEnv",
	"SQLObservation",
	"SQLAction",
	"SQLReward",
	"SQLTask",
	"StepResult",
	"ALL_TASKS",
	"TASK_SETS",
	]