From 158afe1ae548d8511f4f9dba58678c6340ce9325 Mon Sep 17 00:00:00 2001 From: steffen-illium Date: Tue, 18 May 2021 14:43:59 +0200 Subject: [PATCH] first steps to gym dirt spawn frequency --- environments/factory/base_factory.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/environments/factory/base_factory.py b/environments/factory/base_factory.py index 85cb6e2..103e75f 100644 --- a/environments/factory/base_factory.py +++ b/environments/factory/base_factory.py @@ -55,12 +55,13 @@ class BaseFactory(gym.Env): 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)}} + if not self.__class__.__subclasses__(): + self.reset() + else: + self.register_additional_actions() - self.reset() - - def register_actions(self, n_actions): - self._registered_actions += n_actions - return True + def register_additional_actions(self): + raise NotImplementedError('Please register additional actions ') def reset(self) -> (np.ndarray, int, bool, dict): self.done = False