使用 PyEnvironment TF-Agents 包装自定义健身房

Question

我有一个自定义健身房环境，它有一个渲染方法，我可以用go_env.render(mode="human") （它绘制一个 pyglet 画布）调用。 我的健身房适用于go_env = gym.make('gym_go:go-v0', size=args.boardsize, komi=args.komi) 。 我将如何在 TF-Agents 中编写相应的环境包装器？ 目前我有： env = suite_gym.load('gym_go:go-v0', gym_kwargs={'size':3,'komi':0}, render_kwargs={'mode':'terminal'}) 。 但这会抛出 TypeError，因为 PyEnvironment 上的 render() 有它自己的模式：

render(
    mode: Text = 'rgb_array'
) -> Optional[types.NestedArray]

如何使用我的参数正确包装健身房环境，并确保它正确呈现？

Answer 1

原来你不需要在renderkwargs中传递它，你可以像这样将渲染模式直接传递到包装的 class 中：

env = suite_gym.load('gym_go:go-v0', gym_kwargs={'size':3,'komi':0})
env.render('terminal')

这也适用于自定义模式，只要您在自定义健身房实现中重写渲染方法即可。