[core][dataset] Fix typo

darfix/core/dataset.py

@@ -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)