make_fn.py

MAKE_FNS = {
    'breakout':      lambda: lazy_import('pufferlib.ocean.breakout.breakout', 'Breakout'),
    'pong':          lambda: lazy_import('pufferlib.ocean.pong.pong', 'Pong'),
    'enduro':        lambda: lazy_import('pufferlib.ocean.enduro.enduro', 'Enduro'),
    'moba':          lambda: lazy_import('pufferlib.ocean.moba.moba', 'Moba'),
    'nmmo3':         lambda: lazy_import('pufferlib.ocean.nmmo3.nmmo3', 'NMMO3'),
    'snake':         lambda: lazy_import('pufferlib.ocean.snake.snake', 'Snake'),
    'squared':       lambda: lazy_import('pufferlib.ocean.squared.squared', 'Squared'),
    'pysquared':     lambda: lazy_import('pufferlib.ocean.squared.pysquared', 'PySquared'),
    'connect4':      lambda: lazy_import('pufferlib.ocean.connect4.connect4', 'Connect4'),
    'tripletriad':   lambda: lazy_import('pufferlib.ocean.tripletriad.tripletriad', 'TripleTriad'),
    'tactical':      lambda: lazy_import('pufferlib.ocean.tactical.tactical', 'Tactical'),
    'go':            lambda: lazy_import('pufferlib.ocean.go.go', 'Go'),
    'rware':         lambda: lazy_import('pufferlib.ocean.rware.rware', 'Rware'),
    'trash_pickup':  lambda: lazy_import('pufferlib.ocean.trash_pickup.trash_pickup', 'TrashPickupEnv'),
    'tower_climb':   lambda: lazy_import('pufferlib.ocean.tower_climb.tower_climb', 'TowerClimb'),
    'grid':          lambda: lazy_import('pufferlib.ocean.grid.grid', 'Grid'),
    'cpr':           lambda: lazy_import('pufferlib.ocean.cpr.cpr', 'PyCPR'),
    #'rocket_lander': rocket_lander.RocketLander,
    'foraging': make_foraging,
    'predator_prey': make_predator_prey,
    'group': make_group,
    'puffer': make_puffer,
    'continuous': make_continuous,
    'bandit': make_bandit,
    'memory': make_memory,
    'password': make_password,
    'stochastic': make_stochastic,
    'multiagent': make_multiagent,
    'spaces': make_spaces,
    'performance': make_performance,
    'performance_empiric': make_performance_empiric,
}