set_auto_unwrap_transformed_env

class torchrl.set_auto_unwrap_transformed_env(mode: bool)

一个上下文管理器或装饰器，用于控制 TransformedEnv 是否应自动解开嵌套的 TransformedEnv 实例。

参数:

mode (bool) – 是否自动解开嵌套的 TransformedEnv 实例。如果为 False，则 TransformedEnv 将不会解开嵌套实例。默认为 True。

注意

直到 v0.9 版本，如果 TransformedEnv 被嵌套且值未显式设置（auto_unwrap=True 默认行为）时，将引发警告。您可以通过以下方式设置 auto_unwrap_transformed_env() 的值：

• 环境变量 AUTO_UNWRAP_TRANSFORMED_ENV；

• 在脚本开头设置 torchrl.set_auto_unwrap_transformed_env(val: bool).set()；

• 通过将 torchrl.set_auto_unwrap_transformed_env(val: bool) 用作上下文管理器或装饰器。

另请参阅

TransformedEnv

示例

>>> with set_auto_unwrap_transformed_env(False):
...     env = TransformedEnv(TransformedEnv(env))
...     assert not isinstance(env.base_env, TransformedEnv)
>>> @set_auto_unwrap_transformed_env(False)
... def my_function():
...     env = TransformedEnv(TransformedEnv(env))
...     assert not isinstance(env.base_env, TransformedEnv)
...     return env