steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-05-22 14:56:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

added list of maps that require masking (in factory)

Browse Source
This commit is contained in:
Chanumask 2024-09-30 10:09:47 +02:00
parent 2eee415015
commit 95749d8238
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
marl_factory_grid/environment/factory.py
View File

@ -100,6 +100,7 @@ class Factory(gym.Env):
parsed_entities = self.conf.load_entities()
self.map = LevelParser(self.level_filepath, parsed_entities, self.conf.pomdp_r)
self.levels_that_require_masking = ['two_rooms']
# Init for later usage:
# noinspection PyTypeChecker
@ -279,7 +280,7 @@ class Factory(gym.Env):
render_entities = self.filter_entities(render_entities)
# Mask entities based on dynamic conditions instead of hardcoding level-specific logic
if self.conf['General']['level_name'] == 'two_rooms':
if self.conf['General']['level_name'] in self.levels_that_require_masking:
render_entities = self.mask_entities(render_entities)
if self.conf.pomdp_r: