Revision e1467a79dc6580ae009d827b5e6f274faff3b339 authored by liqunfu on 27 March 2020, 21:42:04 UTC, committed by GitHub on 27 March 2020, 21:42:04 UTC
support Pooling ops with Sequence axis
env_factory.py
# Copyright (c) Microsoft. All rights reserved.
# Licensed under the MIT license. See LICENSE.md file in the project root
# for full license information.
# ==============================================================================
from gym import envs
from . import maze2d, puddleworld
def register_env(env_id):
if env_id == 'Maze2D-v0':
envs.register(
id=env_id,
entry_point='env:maze2d.Maze2D',
kwargs={},
max_episode_steps=200,
reward_threshold=-110.0)
elif env_id == 'PuddleWorld-v0':
envs.register(
id=env_id,
entry_point='env:puddleworld.PuddleWorld',
kwargs={},
max_episode_steps=200,
reward_threshold=-100.0)
else:
raise ValueError('Cannot find environment "{0}"\n'.format(env_id))
return True
Computing file changes ...