Spaces:
Sleeping
Sleeping
Commit ·
bdec1e0
1
Parent(s): 5e910ae
Fix reset - handle empty body, task_id defaults to 1
Browse files
main.py
CHANGED
|
@@ -20,7 +20,7 @@ class StepResponse(BaseModel):
|
|
| 20 |
info: dict
|
| 21 |
|
| 22 |
class ResetRequest(BaseModel):
|
| 23 |
-
task_id: int
|
| 24 |
|
| 25 |
class ResetResponse(BaseModel):
|
| 26 |
observation: dict
|
|
@@ -327,7 +327,9 @@ def compute_reward(agent_rows, agent_cols):
|
|
| 327 |
return details["total_reward"], details
|
| 328 |
|
| 329 |
@app.post("/reset", response_model=ResetResponse)
|
| 330 |
-
def reset(req: ResetRequest):
|
|
|
|
|
|
|
| 331 |
if req.task_id not in TASKS:
|
| 332 |
raise HTTPException(status_code=400, detail="task_id must be 1–8")
|
| 333 |
session["task_id"] = req.task_id
|
|
|
|
| 20 |
info: dict
|
| 21 |
|
| 22 |
class ResetRequest(BaseModel):
|
| 23 |
+
task_id: int=1
|
| 24 |
|
| 25 |
class ResetResponse(BaseModel):
|
| 26 |
observation: dict
|
|
|
|
| 327 |
return details["total_reward"], details
|
| 328 |
|
| 329 |
@app.post("/reset", response_model=ResetResponse)
|
| 330 |
+
def reset(req: ResetRequest = None):
|
| 331 |
+
if req is None:
|
| 332 |
+
req = ResetRequest(task_id=1)
|
| 333 |
if req.task_id not in TASKS:
|
| 334 |
raise HTTPException(status_code=400, detail="task_id must be 1–8")
|
| 335 |
session["task_id"] = req.task_id
|