Spaces:
Sleeping
Sleeping
File size: 647 Bytes
3f7c971 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from pettingzoo.utils.conversions import parallel_wrapper_fn
from ._mpe_utils.simple_env import SimpleEnv, make_env
from .scenarios.simple_spread import Scenario
class raw_env(SimpleEnv):
def __init__(self, N=3, local_ratio=0.5, max_cycles=25, continuous_actions=False):
assert 0. <= local_ratio <= 1., "local_ratio is a proportion. Must be between 0 and 1."
scenario = Scenario()
world = scenario.make_world(N)
super().__init__(scenario, world, max_cycles, continuous_actions, local_ratio)
self.metadata['name'] = "simple_spread_v2"
env = make_env(raw_env)
parallel_env = parallel_wrapper_fn(env)
|