Spaces:
Runtime error
Runtime error
prashasti commited on
Commit ·
15c4c1b
1
Parent(s): 6c0967c
App.py changes
Browse files- 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 |
-
|
|
|
|
|
|
|
| 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 {
|