set_gym_backend

torchrl.envs.set_gym_backend(backend)

將 gym-backend 設定為特定值。

參數:

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'>