Spaces:
Sleeping
Sleeping
fix: add state() method
Browse files
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)
|