hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do not check bounds since it is always valid when everything is fine. (#95)

Browse Source
This commit is contained in:
Alexis DUBURCQ 2020-06-25 15:06:35 +02:00 committed by GitHub
parent 3086b5c31d
commit 5ac9f9b144
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tianshou/data/batch.py
View File

@ -171,18 +171,12 @@ class Batch:
b[k] = Batch()
elif hasattr(v, '__len__') and (not isinstance(
v, (np.ndarray, torch.Tensor)) or v.ndim > 0):
if _valid_bounds(len(v), index):
if isinstance(index, (int, np.integer)) or \
(isinstance(index, np.ndarray) and
index.ndim == 0) or \
not isinstance(v, list):
b[k] = v[index]
else:
b[k] = [v[i] for i in index]
if isinstance(index, (int, np.integer)) or \
(isinstance(index, np.ndarray) and
index.ndim == 0) or not isinstance(v, list):
b[k] = v[index]
else:
raise IndexError(
f"Index {index} out of bounds for {type(v)} of "
f"len {len(self)}.")
b[k] = [v[i] for i in index]
return b
def __setitem__(self, index: Union[