steffen/FIKS_Entwicklungsumgebung
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Item debugging and New Entities

Browse Source
This commit is contained in:
Steffen Illium
2021-08-23 18:39:50 +02:00
parent d5e4d44823
commit c3d4925653
4 changed files with 43 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
environments/factory/base/base_factory.py
View File

@ -157,7 +157,7 @@ class BaseFactory(gym.Env):
entities.register_additional_items([self._doors])
if additional_entities := self.additional_entities:
entities.register_additional_items([additional_entities])
entities.register_additional_items(additional_entities)
return entities

6
environments/factory/base/registers.py
View File

@ -93,6 +93,10 @@ class Register:
class EntityRegister(Register):
@property
def positions(self):
return [agent.pos for agent in self]
def __init__(self):
super(EntityRegister, self).__init__()
self._tiles = dict()
@ -150,7 +154,7 @@ class FloorTiles(EntityRegister):
return tiles
class Agents(Register):
class Agents(EntityRegister):
_accepted_objects = Agent