samrat-rm commited on
Commit
3613ecf
·
1 Parent(s): 243b472

fix: add state() method

Browse files
Files changed (1) hide show
  1. server/WhyDidItFail_environment.py +4 -0
server/WhyDidItFail_environment.py CHANGED
@@ -27,6 +27,10 @@ class WhyDidItFailEnvironment(Environment):
27
  self.scenario = None
28
  self.inspected = set() # tracks what the agent has already looked at TODO : implement inspected logic
29
 
 
 
 
 
30
  def reset(self, seed: Optional[int] = None, episode_id: Optional[str] = None, **kwargs: Any) -> WhyDidItFailObservation:
31
  if seed is not None:
32
  random.seed(seed)
 
27
  self.scenario = None
28
  self.inspected = set() # tracks what the agent has already looked at TODO : implement inspected logic
29
 
30
+ @property
31
+ def state(self) -> State:
32
+ return self._state
33
+
34
  def reset(self, seed: Optional[int] = None, episode_id: Optional[str] = None, **kwargs: Any) -> WhyDidItFailObservation:
35
  if seed is not None:
36
  random.seed(seed)