steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-05-23 07:16:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

first steps to gym

Browse Source 
dirt spawn frequency
This commit is contained in:
steffen-illium 2021-05-18 14:43:59 +02:00
parent 5a9a3257ed
commit 158afe1ae5
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
environments/factory/base_factory.py
View File

@ -55,12 +55,13 @@ class BaseFactory(gym.Env):
h.parse_level(Path(__file__).parent / h.LEVELS_DIR / f'{level}.txt') h.parse_level(Path(__file__).parent / h.LEVELS_DIR / f'{level}.txt')
) )
self.slice_strings = {0: 'level', **{i: f'agent#{i}' for i in range(1, self.n_agents+1)}} self.slice_strings = {0: 'level', **{i: f'agent#{i}' for i in range(1, self.n_agents+1)}}
if not self.__class__.__subclasses__():
self.reset()
else:
self.register_additional_actions()
self.reset() def register_additional_actions(self):
raise NotImplementedError('Please register additional actions ')
def register_actions(self, n_actions):
self._registered_actions += n_actions
return True
def reset(self) -> (np.ndarray, int, bool, dict): def reset(self) -> (np.ndarray, int, bool, dict):
self.done = False self.done = False