Spaces:
Sleeping
Sleeping
File size: 511 Bytes
3f7c971 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from pettingzoo.utils.conversions import parallel_wrapper_fn
from ._mpe_utils.simple_env import SimpleEnv, make_env
from .scenarios.simple_push import Scenario
class raw_env(SimpleEnv):
def __init__(self, max_cycles=25, continuous_actions=False):
scenario = Scenario()
world = scenario.make_world()
super().__init__(scenario, world, max_cycles, continuous_actions)
self.metadata['name'] = "simple_push_v2"
env = make_env(raw_env)
parallel_env = parallel_wrapper_fn(env)
|