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