Tf Agents Parallel Py Environment со средой, которая имеет входные параметры

Question

Предположим, у вас есть среда, в которой есть входные параметры: например, для создания экземпляра вы должны использовать

env_instance = MyEnv(var_1=3, var_2=5, ...)

Теперь предположим, что вы хотите создать Parallel_py_environment, используя среду " MyEnv "? Поскольку вам нужны входные параметры, вы не можете использовать

tf_py_environment.TFPyEnvironment(parallel_py_environment.ParallelPyEnvironment([MyEnv]*int(n_envs)))

Answer 1

Решением является создание суперкласса:

class MyEnvPar(MyEnv):
    def __init__(self):
        super().__init__(var_1=3, var_2=5)

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

tf_py_environment.TFPyEnvironment(parallel_py_environment.ParallelPyEnvironment([MyEnvPar]*int(n_envs)))