Farama-Foundation / MAgent2

An engine for high performance multi-agent environments with very large numbers of agents, along with a set of reference environments
https://magent2.farama.org
MIT License
201 stars 34 forks source link

MAgent2 #1

Closed dsctt closed 1 year ago

dsctt commented 1 year ago
from magent2.environments import battle_v4, adversarial_pursuit_v4, battlefield_v5, combined_arms_v6, gather_v5
from pettingzoo.utils import random_demo

env = battle_v4.env(map_size=16, minimap_mode=False, step_reward=-0.005,
                    dead_penalty=-0.1, attack_penalty=-0.1, attack_opponent_reward=0.2,
                    max_cycles=100000, extra_features=False, render_mode='human')

# env = adversarial_pursuit_v4.env(map_size=12, minimap_mode=False, 
#                     max_cycles=100000, extra_features=False, render_mode='human')

# env = battlefield_v5.env(map_size=46, minimap_mode=False, 
#                     max_cycles=100000, extra_features=False, render_mode='human')

# env = combined_arms_v6.env(map_size=46, minimap_mode=False, 
#                     max_cycles=100000, extra_features=False, render_mode='human')

# env = gather_v5.env(minimap_mode=False, 
#                     max_cycles=100000, extra_features=False, render_mode='human')

env.reset()
random_demo(env, render=True, episodes=1)