Fix 311 bug (#1426)

albumentations-team · Mar 23, 2023 · 16cbc53 · 16cbc53
1 parent 9b0525f
commit 16cbc53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/albumentations/augmentations/blur/transforms.py b/albumentations/augmentations/blur/transforms.py
@@ -41,7 +41,7 @@ def apply(self, img: np.ndarray, ksize: int = 3, **params) -> np.ndarray:
         return F.blur(img, ksize)
 
     def get_params(self) -> Dict[str, Any]:
-        return {"ksize": int(random.choice(np.arange(self.blur_limit[0], self.blur_limit[1] + 1, 2)))}
+        return {"ksize": int(random.choice(list(range(self.blur_limit[0], self.blur_limit[1] + 1, 2))))}
 
     def get_transform_init_args_names(self) -> Tuple[str, ...]:
         return ("blur_limit",)
@@ -84,7 +84,7 @@ def apply(self, img: np.ndarray, kernel: np.ndarray = None, **params) -> np.ndar
         return FMain.convolve(img, kernel=kernel)
 
     def get_params(self) -> Dict[str, Any]:
-        ksize = random.choice(np.arange(self.blur_limit[0], self.blur_limit[1] + 1, 2))
+        ksize = random.choice(list(range(self.blur_limit[0], self.blur_limit[1] + 1, 2)))
         if ksize <= 2:
             raise ValueError("ksize must be > 2. Got: {}".format(ksize))
         kernel = np.zeros((ksize, ksize), dtype=np.uint8)