feat: fix bug

web_server.py

@@ -127,11 +127,13 @@ def update_submission_status(team_id: str, submission_id: str, status: int):
 
 
 def delete_client_space(client_space_id: str):
-    return
-    hf_api.delete_repo(
-        repo_id=client_space_id,
-        repo_type="space"
-    )
+    try:
+        hf_api.delete_repo(
+            repo_id=client_space_id,
+            repo_type="space"
+        )
+    except:
+        print(f"Failed to delete space {client_space_id}. It may not exist or already deleted.")
 
 
 class FifoDict:
@@ -313,6 +315,7 @@ class EnvHandlerManager:
         """
         with self._lock:
             env = self._env_handlers.pop(env_id, None)
+            self._env_handlers[env_id] = None
         if env is not None:
             env.close()
         torch.cuda.empty_cache()
@@ -335,7 +338,10 @@ def _get_env_handler(
         token_info = get_token_info(token)
     except Exception:
         raise HTTPException(status_code=401)
-    return env_manager.get_env_handler(token_info["submission_id"])
+    env_handler = env_manager.get_env_handler(token_info["submission_id"])
+    if env_handler is None:
+        raise HTTPException(status_code=404, detail="Environment handler already closed.")
+    return env_handler
 
 
 def _load_numpy_ndarray_json_str(json_str: str) -> np.ndarray:
@@ -385,7 +391,7 @@ def execute_action_endpoint(
         return Response(content=cache_result, media_type="application/octet-stream")
 
     if env_handler.has_done:
-        result = pickle.dumps({"done": True, "state": None})
+        result = pickle.dumps({"done": True, "state": env_handler.get_current_state()})
         _result_dict.push(transaction_id, result)
         return Response(content=result, media_type="application/octet-stream")
 
@@ -402,7 +408,7 @@ def execute_action_endpoint(
             repo_type="dataset",
             path_in_repo=f"eval_results/{token_info['submission_id']}",
         )
-        result = pickle.dumps({"done": done, "state": None})
+        result = pickle.dumps({"done": done, "state": env_handler.get_current_state()})
         _result_dict.push(transaction_id, result)
         return Response(content=result, media_type="application/octet-stream")