prashasti commited on
Commit
15c4c1b
·
1 Parent(s): 6c0967c

App.py changes

Browse files
Files changed (1) hide show
  1. server/app.py +4 -2
server/app.py CHANGED
@@ -65,10 +65,12 @@ def health():
65
 
66
 
67
  @app.post("/reset")
68
- def reset(request: ResetRequest):
69
  global _env
70
  try:
71
- _env = create_env(request.task)
 
 
72
  obs = _env.reset()
73
 
74
  return {
 
65
 
66
 
67
  @app.post("/reset")
68
+ def reset(request: Optional[ResetRequest] = None):
69
  global _env
70
  try:
71
+ task = request.task if request else "simple"
72
+
73
+ _env = create_env(task)
74
  obs = _env.reset()
75
 
76
  return {