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

Restructuring

Browse Source
This commit is contained in:
steffen-illium
2021-06-17 16:23:35 +02:00
parent 7b4060a042
commit 7893e1131e
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
algorithms/_base.py
View File

@ -209,7 +209,7 @@ if __name__ == '__main__':
allow_square_movement=True,
allow_no_op=False)
env = SimpleFactory(dirt_properties=dirt_props, movement_properties=move_props, n_agents=N_AGENTS, pomdp_radius=2, max_steps=400, omit_agent_slice_in_obs=False)
env = DummyVecEnv([lambda: env])
# env = DummyVecEnv([lambda: env])
print(env)
from stable_baselines3.dqn import DQN