Spaces:
Sleeping
Sleeping
Commit ·
cf77682
1
Parent(s): f73e70d
Make reset body optional for OpenEnv validator
Browse files- email_env/server/app.py +4 -2
email_env/server/app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from fastapi import FastAPI, HTTPException
|
| 2 |
from pydantic import BaseModel
|
| 3 |
|
|
@@ -32,9 +33,10 @@ class GradeRequest(BaseModel):
|
|
| 32 |
|
| 33 |
|
| 34 |
@app.post("/reset", response_model=Observation)
|
| 35 |
-
def reset(req: ResetRequest):
|
|
|
|
| 36 |
try:
|
| 37 |
-
obs = env.reset(task_id=
|
| 38 |
except ValueError as e:
|
| 39 |
raise HTTPException(status_code=400, detail=str(e))
|
| 40 |
return obs
|
|
|
|
| 1 |
+
from typing import Optional
|
| 2 |
from fastapi import FastAPI, HTTPException
|
| 3 |
from pydantic import BaseModel
|
| 4 |
|
|
|
|
| 33 |
|
| 34 |
|
| 35 |
@app.post("/reset", response_model=Observation)
|
| 36 |
+
def reset(req: Optional[ResetRequest] = None):
|
| 37 |
+
task_id = req.task_id if req else "task_1"
|
| 38 |
try:
|
| 39 |
+
obs = env.reset(task_id=task_id)
|
| 40 |
except ValueError as e:
|
| 41 |
raise HTTPException(status_code=400, detail=str(e))
|
| 42 |
return obs
|