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

environment code changes for RL settings

Browse Source
This commit is contained in:
Julian Schönberger
2024-05-02 10:54:46 +02:00
parent 50bcf5d995
commit 48d708bbcd
3 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
marl_factory_grid/algorithms/utils.py
View File

@@ -65,8 +65,12 @@ def add_env_props(cfg):
_ = factory.reset()
# Agent Init
cfg['agent'].update(dict(observation_size=list(factory.observation_space[0].shape),
n_actions=factory.action_space[0].n))
if len(factory.state.moving_entites) == 1: # Single agent setting
observation_size = list(factory.observation_space.shape)
else: # Multi-agent setting
observation_size = list(factory.observation_space[0].shape)
cfg['agent'].update(dict(observation_size=observation_size, n_actions=factory.action_space[0].n))
return factory