register_env.py

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,
)