gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-07-30 21:34:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add generic conversion wrapper between Array API compatible frameworks (#1333)

Browse Source
This commit is contained in:
Martin Schuck
2025-05-12 00:10:06 +02:00
committed by GitHub
parent 95637ebc7f
commit 5dde9a79be
23 changed files with 1039 additions and 623 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/api/wrappers/misc_wrappers.md
View File

@@ -27,6 +27,7 @@ title: Misc Wrappers
## Data Conversion Wrappers
```{eval-rst}
.. autoclass:: gymnasium.wrappers.ArrayConversion
.. autoclass:: gymnasium.wrappers.JaxToNumpy
.. autoclass:: gymnasium.wrappers.JaxToTorch
.. autoclass:: gymnasium.wrappers.NumpyToTorch

2
docs/api/wrappers/table.md
View File

@@ -34,6 +34,8 @@ wrapper in the page on the wrapper type
- Converts an image observation computed by ``reset`` and ``step`` from RGB to Grayscale.
* - :class:`HumanRendering`
- Allows human like rendering for environments that support "rgb_array" rendering.
* - :class:`ArrayConversion`
- Wraps an environment based on any Array API compatible framework, e.g. torch, jax, numpy, such that it can be interacted with any other Array API compatible framework.
* - :class:`JaxToNumpy`
- Wraps a Jax-based environment such that it can be interacted with NumPy arrays.
* - :class:`JaxToTorch`