gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-16 03:38:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

wrappers.vector.NumpyToTorch uses Device from wrappers.NumpyTorch not JaxToTorch (#1308)

Browse Source
This commit is contained in:
Mark Towers
2025-02-13 22:45:30 +00:00
committed by GitHub
parent 2a8441d439
commit a8946325e8
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gymnasium/wrappers/numpy_to_torch.py
View File

@@ -24,7 +24,7 @@ except ImportError:
)
__all__ = ["NumpyToTorch", "torch_to_numpy", "numpy_to_torch"]
__all__ = ["NumpyToTorch", "torch_to_numpy", "numpy_to_torch", "Device"]
# The NoneType is not defined in Python 3.9. Remove when the minimal version is bumped to >=3.10
_NoneType = type(None)