Fix mypy errors (#893)

SeldonIO · May 21, 2024 · 39fff84 · 39fff84
1 parent c1297e8
commit 39fff84
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/alibi_detect/utils/perturbation.py b/alibi_detect/utils/perturbation.py
@@ -865,8 +865,8 @@ def pixelate(x: np.ndarray, strength: float, xrange: tuple = None) -> np.ndarray
  x = (x - xrange[0]) / (xrange[1] - xrange[0]) * 255
 
  im = Image.fromarray(x.astype('uint8'), mode='RGB')
- im = im.resize((int(rows * strength), int(cols * strength)), Image.BOX)
- im = im.resize((rows, cols), Image.BOX)
+ im = im.resize((int(rows * strength), int(cols * strength)), Image.Resampling.BOX)
+ im = im.resize((rows, cols), Image.Resampling.BOX)
  x_pi = np.array(im, dtype=np.float32) / 255
  x_pi = x_pi * (xrange[1] - xrange[0]) + xrange[0]
  return x_pi
@@ -898,8 +898,8 @@ def jpeg_compression(x: np.ndarray, strength: float, xrange: tuple = None) -> np
  x = Image.fromarray(x.astype('uint8'), mode='RGB')
  output = BytesIO()
  x.save(output, 'JPEG', quality=strength) # type: ignore[attr-defined] # TODO: allow redefinition
- x = Image.open(output)
- x_jpeg = np.array(x, dtype=np.float32) / 255
+ x_image = Image.open(output)
+ x_jpeg = np.array(x_image, dtype=np.float32) / 255
  x_jpeg = x_jpeg * (xrange[1] - xrange[0]) + xrange[0]
  return x_jpeg