Fix FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use arr[tuple(seq)] instead of arr[seq]. In the future this will be interpreted as an array index, arr[np.array(seq)], which will result either in an error or a different result. (#6568)

This commit is contained in:
mehrdadn
2019-12-22 13:02:34 -08:00
committed by Robert Nishihara
parent 7bbfa85c66
commit 50fb26de68
2 changed files with 6 additions and 5 deletions
+2 -1
View File
@@ -47,7 +47,8 @@ def hstack(*xs):
# TODO(rkn): Be consistent about using "index" versus "indices".
@ray.remote
def subarray(a, lower_indices, upper_indices):
return a[[slice(l, u) for (l, u) in zip(lower_indices, upper_indices)]]
idx = tuple(slice(l, u) for (l, u) in zip(lower_indices, upper_indices))
return a[idx]
@ray.remote