Agent_Control_with_Language / register_env.py
ArseniyPerchik's picture
more
a853d77
raw
history blame contribute delete
457 Bytes
from gymnasium.envs.registration import register
from warehouse_env import WarehouseEnv
# Example for the CartPole environment
register(
# unique identifier for the env `name-version`
id="WarehouseEnv",
# path to the class for creating the env
# Note: entry_point also accept a class as input (and not only a string)
entry_point=WarehouseEnv,
# Max number of steps per episode, using a `TimeLimitWrapper`
max_episode_steps=500,
)