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(): ...@@ -797,7 +797,7 @@ class Dataset():
if not self.operations_state[Operation.SHIFT]: if not self.operations_state[Operation.SHIFT]:
return return
filename = _dir + "/data" + str(i).zfill(4) + ".npy" filename = _dir + "/data" + str(i).zfill(4) + ".npy"
if shift[:, i] > 1: if shift[:, i].all() > 1:
shift_approach = "linear" shift_approach = "linear"
img = apply_shift(data[i], shift[:, i], shift_approach) img = apply_shift(data[i], shift[:, i], shift_approach)
numpy.save(filename, img) numpy.save(filename, img)
...@@ -830,7 +830,7 @@ class Dataset(): ...@@ -830,7 +830,7 @@ class Dataset():
if not self.operations_state[Operation.SHIFT]: if not self.operations_state[Operation.SHIFT]:
return return
filename = _dir + "/data" + str(i).zfill(4) + ".npy" filename = _dir + "/data" + str(i).zfill(4) + ".npy"
if shift[:, i] > 1: if shift[:, i].all() > 1:
shift_approach = "linear" shift_approach = "linear"
img = apply_shift(data[i], shift[:, i], shift_approach) img = apply_shift(data[i], shift[:, i], shift_approach)
numpy.save(filename, img) numpy.save(filename, img)
......
Supports Markdown
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