ritvik360 commited on
Commit
c071eb6
·
verified ·
1 Parent(s): 25575d3

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -171,7 +171,7 @@ async def run_task(client: OpenAI, env, task_name: str) -> dict:
171
  # OpenEnv reset() may not accept task args via HTTP; we rely on
172
  # NL2SQL_DEFAULT_TASK env-var being set before calling, OR we
173
  # pass it as a reset parameter if the server supports it.
174
- result = await env.reset(options={"task": task_name}) # changed
175
  obs = result.observation
176
 
177
  for step in range(1, MAX_STEPS + 1):
 
171
  # OpenEnv reset() may not accept task args via HTTP; we rely on
172
  # NL2SQL_DEFAULT_TASK env-var being set before calling, OR we
173
  # pass it as a reset parameter if the server supports it.
174
+ result = await env.reset() # changed
175
  obs = result.observation
176
 
177
  for step in range(1, MAX_STEPS + 1):