Использование пользовательских сред OpenAI

Question

У меня есть файл rb.py, который примерно структурирован следующим образом:

import gym

def main():
   env = gym.make('ComplexMiniGolf-v0')

Проблема в том, что если я создам другой файл и сделаю:

from rb import main
main()

Пользовательская среда загружается и все работает. Но если вместо этого:

import gym
env = gym.make('ComplexMiniGolf-v0')

Пользовательское окружение не может быть найдено, даже если они загружают один и тот же пакет.

Answer 1

Зарегистрировали ли вы свою среду в тренажерном зале?

Вы могли бы сделать это таким образом, если бы ваша собственная среда была определена в следующей структуре:

myenv/
    __init__.py
    myenv.py

myenv.py содержит класс для ваше окружение. В __init__.py вы вводите следующий код:

from gym.envs.registration import register

register(
    id='MyEnv-v0',
    entry_point='myenv.myenv:MyEnv',
)

Тогда вы можете использовать его так:

import gym
import myenv
env = gym.make('MyEnv-v0')