P-Karthik-Mohan commited on
Commit
bdec1e0
·
1 Parent(s): 5e910ae

Fix reset - handle empty body, task_id defaults to 1

Browse files
Files changed (1) hide show
  1. main.py +4 -2
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