gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-25 15:59:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix non-used device argument in to_torch conversion (#1107)

Browse Source
This commit is contained in:
Mantas
2024-07-03 10:16:52 +01:00
committed by GitHub
parent 57136fbaee
commit 6b22fd5534
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gymnasium/wrappers/jax_to_torch.py
View File

@@ -119,9 +119,9 @@ def _jax_iterable_to_torch(
if hasattr(value, "_make"): if hasattr(value, "_make"):
# namedtuple - underline used to prevent potential name conflicts # namedtuple - underline used to prevent potential name conflicts
# noinspection PyProtectedMember # noinspection PyProtectedMember
return type(value)._make(jax_to_torch(v) for v in value) return type(value)._make(jax_to_torch(v, device) for v in value)
else: else:
return type(value)(jax_to_torch(v) for v in value) return type(value)(jax_to_torch(v, device) for v in value)
class JaxToTorch(gym.Wrapper, gym.utils.RecordConstructorArgs): class JaxToTorch(gym.Wrapper, gym.utils.RecordConstructorArgs):

4
gymnasium/wrappers/numpy_to_torch.py
View File

@@ -94,9 +94,9 @@ def _numpy_iterable_to_torch(
if hasattr(value, "_make"): if hasattr(value, "_make"):
# namedtuple - underline used to prevent potential name conflicts # namedtuple - underline used to prevent potential name conflicts
# noinspection PyProtectedMember # noinspection PyProtectedMember
return type(value)._make(numpy_to_torch(v) for v in value) return type(value)._make(numpy_to_torch(v, device) for v in value)
else: else:
return type(value)(numpy_to_torch(v) for v in value) return type(value)(numpy_to_torch(v, device) for v in value)
class NumpyToTorch(gym.Wrapper, gym.utils.RecordConstructorArgs): class NumpyToTorch(gym.Wrapper, gym.utils.RecordConstructorArgs):