Commit ee9daef4 authored by Julia Garriga Ferrer's avatar Julia Garriga Ferrer
Browse files

[core][dataset] Fix typo

parent 9141451d
......@@ -797,7 +797,7 @@ class Dataset():
if not self.operations_state[Operation.SHIFT]:
return
filename = _dir + "/data" + str(i).zfill(4) + ".npy"
if shift[:, i] > 1:
if shift[:, i].all() > 1:
shift_approach = "linear"
img = apply_shift(data[i], shift[:, i], shift_approach)
numpy.save(filename, img)
......@@ -830,7 +830,7 @@ class Dataset():
if not self.operations_state[Operation.SHIFT]:
return
filename = _dir + "/data" + str(i).zfill(4) + ".npy"
if shift[:, i] > 1:
if shift[:, i].all() > 1:
shift_approach = "linear"
img = apply_shift(data[i], shift[:, i], shift_approach)
numpy.save(filename, img)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment