sravaniamere commited on
Commit
f6bd747
·
1 Parent(s): 66cc86e

switch to create_app from http_server

Browse files
Files changed (1) hide show
  1. 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 create_fastapi_app
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