Libra-1995 commited on
Commit
1102668
·
1 Parent(s): d2730b8

feat: current_state return done

Browse files
Files changed (1) hide show
  1. web_server.py +1 -1
web_server.py CHANGED
@@ -255,7 +255,7 @@ class WebServer:
255
 
256
  def _get_current_state_endpoint(self):
257
  state = self.env_handler.get_current_state()
258
- return Response(content=pickle.dumps(state), media_type="application/octet-stream")
259
 
260
  def _load_numpy_ndarray_json_str(self, json_str: str) -> np.ndarray:
261
  """
 
255
 
256
  def _get_current_state_endpoint(self):
257
  state = self.env_handler.get_current_state()
258
+ return Response(content=pickle.dumps({"done": self.env_handler.has_done, "state": state}), media_type="application/octet-stream")
259
 
260
  def _load_numpy_ndarray_json_str(self, json_str: str) -> np.ndarray:
261
  """