hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

avoid DeprecationWarning

Browse Source
This commit is contained in:
NM512
2023-06-18 17:18:24 +09:00
parent e3329b35e5
commit b408067d9a
4 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
envs/memorymaze.py
View File

@@ -36,9 +36,9 @@ class MemoryMaze:
return {
**spaces,
"reward": gym.spaces.Box(-np.inf, np.inf, (), dtype=np.float32),
"is_first": gym.spaces.Box(0, 1, (), dtype=np.bool),
"is_last": gym.spaces.Box(0, 1, (), dtype=np.bool),
"is_terminal": gym.spaces.Box(0, 1, (), dtype=np.bool),
"is_first": gym.spaces.Box(0, 1, (), dtype=bool),
"is_last": gym.spaces.Box(0, 1, (), dtype=bool),
"is_terminal": gym.spaces.Box(0, 1, (), dtype=bool),
}
@property

4
envs/wrappers.py
View File

@@ -77,8 +77,8 @@ class CollectDataset:
dtype = {16: np.int16, 32: np.int32, 64: np.int64}[self._precision]
elif np.issubdtype(value.dtype, np.uint8):
dtype = np.uint8
elif np.issubdtype(value.dtype, np.bool):
dtype = np.bool
elif np.issubdtype(value.dtype, bool):
dtype = bool
else:
raise NotImplementedError(value.dtype)
return value.astype(dtype)