Spaces:
Running
Running
Commit ·
09576c0
1
Parent(s): bdb64b6
Ensure /step returns info object perfectly matching OpenEnv spec
Browse files- server/app.py +2 -0
server/app.py
CHANGED
|
@@ -226,6 +226,7 @@ async def step(request: Request):
|
|
| 226 |
'reward': round(float(result.get('reward', 0.0)), 4),
|
| 227 |
'done': bool(result.get('done', False)),
|
| 228 |
'observation': step_obs,
|
|
|
|
| 229 |
}
|
| 230 |
except Exception as e:
|
| 231 |
return JSONResponse(status_code=200, content={
|
|
@@ -233,6 +234,7 @@ async def step(request: Request):
|
|
| 233 |
'done': True,
|
| 234 |
'error': str(e),
|
| 235 |
'observation': {},
|
|
|
|
| 236 |
})
|
| 237 |
|
| 238 |
|
|
|
|
| 226 |
'reward': round(float(result.get('reward', 0.0)), 4),
|
| 227 |
'done': bool(result.get('done', False)),
|
| 228 |
'observation': step_obs,
|
| 229 |
+
'info': {'validation_failed': step_obs.get('validation_failed', False)},
|
| 230 |
}
|
| 231 |
except Exception as e:
|
| 232 |
return JSONResponse(status_code=200, content={
|
|
|
|
| 234 |
'done': True,
|
| 235 |
'error': str(e),
|
| 236 |
'observation': {},
|
| 237 |
+
'info': {'error': str(e)},
|
| 238 |
})
|
| 239 |
|
| 240 |
|