Merge remote-tracking branch 'origin/AgentState_Object'

2021-05-14 15:22:18 +02:00
parent ced637a348 83d77df216
commit 6b1b14fa87
2 changed files with 18 additions and 9 deletions
--- a/environments/factory/base_factory.py
+++ b/environments/factory/base_factory.py
@@ -93,7 +93,7 @@ class BaseFactory:
        self.slice_strings = {0: 'level', **{i: f'agent#{i}' for i in range(1, self.n_agents+1)}}
        self.reset()

-    def reset(self):
+    def reset(self)  -> (np.ndarray, int, bool, dict):
        self.done = False
        self.steps = 0
        self.cumulative_reward = 0