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

added test hooks (like rules)

Browse Source
This commit is contained in:
Chanumask
2023-11-10 10:41:41 +01:00
parent 9b9c6e0385
commit 64c0d0e4e9
3 changed files with 84 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
marl_factory_grid/environment/factory.py
View File

@@ -87,16 +87,19 @@ class Factory(gym.Env):
entities = self.map.do_init()
# Init rules
rules = self.conf.load_rules()
rules = self.conf.load_env_rules()
env_tests = self.conf.load_env_tests() if self.conf.tests else []
# Parse the agent conf
parsed_agents_conf = self.conf.parse_agents_conf()
self.state = Gamestate(entities, parsed_agents_conf, rules, self.conf.env_seed, self.conf.verbose)
self.state = Gamestate(entities, parsed_agents_conf, rules, env_tests, self.conf.env_seed, self.conf.verbose)
# All is set up, trigger entity init with variable pos
# All is set up, trigger additional init (after agent entity spawn etc)
self.state.rules.do_all_init(self.state, self.map)
self.state.tests.do_all_init(self.state, self.map)
# Build initial observations for all agents
# noinspection PyAttributeOutsideInit
self.obs_builder = OBSBuilder(self.map.level_shape, self.state, self.map.pomdp_r)