mirror of
https://github.com/ltcptgeneral/cse151b-final-project.git
synced 2024-09-20 08:57:56 +00:00
84 lines
1.6 KiB
Python
84 lines
1.6 KiB
Python
from gym.envs.registration import (
|
|
registry,
|
|
register,
|
|
make,
|
|
spec,
|
|
load_env_plugins as _load_env_plugins,
|
|
)
|
|
|
|
|
|
# Classic
|
|
# ----------------------------------------
|
|
|
|
register(
|
|
id="WordleEnv10-v0",
|
|
entry_point="wordle.wordle:WordleEnv10",
|
|
max_episode_steps=200,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv100-v0",
|
|
entry_point="wordle.wordle:WordleEnv100",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv100OneAction-v0",
|
|
entry_point="wordle.wordle:WordleEnv100OneAction",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv100TwoAction-v0",
|
|
entry_point="wordle.wordle:WordleEnv100TwoAction",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv100FullAction-v0",
|
|
entry_point="wordle.wordle:WordleEnv100FullAction",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv100WithMask-v0",
|
|
entry_point="wordle.wordle:WordleEnv100WithMask",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv1000-v0",
|
|
entry_point="wordle.wordle:WordleEnv1000",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv1000WithMask-v0",
|
|
entry_point="wordle.wordle:WordleEnv1000WithMask",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnv1000FullAction-v0",
|
|
entry_point="wordle.wordle:WordleEnv1000FullAction",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnvFull-v0",
|
|
entry_point="wordle.wordle:WordleEnvFull",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnvReal-v0",
|
|
entry_point="wordle.wordle:WordleEnvReal",
|
|
max_episode_steps=500,
|
|
)
|
|
|
|
register(
|
|
id="WordleEnvRealWithMask-v0",
|
|
entry_point="wordle.wordle:WordleEnvRealWithMask",
|
|
max_episode_steps=500,
|
|
)
|