Gym-MiniGrid/gym_minigrid/register.py

from gym.envs.registration import register as gymRegister

envSet = set()

def register(
    id,
    entry_point,
    reward_threshold=900
):
    assert id.startswith("MiniGrid-")

    # Register the environment with OpenAI gym
    gymRegister(
        id=id,
        entry_point=entry_point,
        reward_threshold=reward_threshold
    )

    # Add the environment to the set
    envSet.add(id)