set_gym_backend

torchrl.envs.set_gym_backend(backend)

將 gym 後端設定為某個值。

引數：

backend (Python 模組、字串或返回模組的可呼叫物件) – 要使用的 gym 後端。如果您想避免在載入時匯入 gym，請使用字串或可呼叫物件。

示例

>>> import gym
>>> import gymnasium
>>> with set_gym_backend("gym"):
...     assert gym_backend() == gym
>>> with set_gym_backend(lambda: gym):
...     assert gym_backend() == gym
>>> with set_gym_backend(gym):
...     assert gym_backend() == gym
>>> with set_gym_backend("gymnasium"):
...     assert gym_backend() == gymnasium
>>> with set_gym_backend(lambda: gymnasium):
...     assert gym_backend() == gymnasium
>>> with set_gym_backend(gymnasium):
...     assert gym_backend() == gymnasium

此函式也可以用作函式裝飾器。

示例

>>> @set_gym_backend("gym")
... def fun():
...     gym = gym_backend()
...     print(gym)
>>> fun()
<module 'gym' from '/path/to/env/site-packages/gym/__init__.py'>
>>> @set_gym_backend("gymnasium")
... def fun():
...     gym = gym_backend()
...     print(gym)
>>> fun()
<module 'gymnasium' from '/path/to/env/site-packages/gymnasium/__init__.py'>