Spaces:
Sleeping
Sleeping
Commit ·
f6bd747
1
Parent(s): 66cc86e
switch to create_app from http_server
Browse files- sql_env/server.py +4 -3
sql_env/server.py
CHANGED
|
@@ -3,7 +3,7 @@ FastAPI server using openenv.core base classes — required for validator.
|
|
| 3 |
"""
|
| 4 |
import random
|
| 5 |
from typing import Optional
|
| 6 |
-
from openenv.core.env_server import
|
| 7 |
from openenv.core.env_server.interfaces import Environment
|
| 8 |
from openenv.core.env_server.types import State
|
| 9 |
|
|
@@ -91,11 +91,12 @@ class SQLCorrectionEnvironment(Environment):
|
|
| 91 |
rewards_history=self._rewards_history,
|
| 92 |
)
|
| 93 |
|
| 94 |
-
|
| 95 |
-
app = create_fastapi_app(
|
| 96 |
SQLCorrectionEnvironment,
|
| 97 |
SQLAction,
|
| 98 |
SQLObservation,
|
|
|
|
|
|
|
| 99 |
)
|
| 100 |
|
| 101 |
|
|
|
|
| 3 |
"""
|
| 4 |
import random
|
| 5 |
from typing import Optional
|
| 6 |
+
from openenv.core.env_server.http_server import create_app
|
| 7 |
from openenv.core.env_server.interfaces import Environment
|
| 8 |
from openenv.core.env_server.types import State
|
| 9 |
|
|
|
|
| 91 |
rewards_history=self._rewards_history,
|
| 92 |
)
|
| 93 |
|
| 94 |
+
app = create_app(
|
|
|
|
| 95 |
SQLCorrectionEnvironment,
|
| 96 |
SQLAction,
|
| 97 |
SQLObservation,
|
| 98 |
+
env_name="sql-correction-env",
|
| 99 |
+
max_concurrent_envs=10,
|
| 100 |
)
|
| 101 |
|
| 102 |
|