Removed debug-prints from StateObservable

This commit is contained in:
Dominik Moritz Roth 2022-06-21 22:30:12 +02:00
parent 983d5071a4
commit 2132deedcd

View File

@ -188,7 +188,6 @@ class StateObservable(Observable):
def entities(self): def entities(self):
if not self._entities == None: if not self._entities == None:
return self._entities return self._entities
print('Building StateGetters')
rewardsWhitelist = self.rewardsWhitelist or self.env.entities rewardsWhitelist = self.rewardsWhitelist or self.env.entities
enemysWhitelist = self.enemysWhitelist or self.env.entities enemysWhitelist = self.enemysWhitelist or self.env.entities
self._entities = [] self._entities = []
@ -207,7 +206,6 @@ class StateObservable(Observable):
for entity in enemysWhitelist: for entity in enemysWhitelist:
if isinstance(entity, entities.TimeoutReward): if isinstance(entity, entities.TimeoutReward):
self._timeoutEntities.append(entity) self._timeoutEntities.append(entity)
print(len(self._entities))
return self._entities return self._entities
def get_observation_space(self): def get_observation_space(self):