Code update

PiperOrigin-RevId: 547905268
google-research · Jul 13, 2023 · 69f320c · 69f320c
1 parent 0b0ed8e
commit 69f320c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/jax3d/projects/generative/common/image_utility.py b/jax3d/projects/generative/common/image_utility.py
@@ -118,9 +118,11 @@ def byte_array_to_image(byte_array: bytes) -> np.ndarray:
   return np.array(Image.open(io.BytesIO(byte_array)))
 
 
-def resize(image_array: np.ndarray,
-           size: Tuple[int, int],
-           resample: int = Image.BICUBIC) -> np.ndarray:
+def resize(
+    image_array: np.ndarray,
+    size: Tuple[int, int],
+    resample: int = Image.Resampling.BICUBIC,
+) -> np.ndarray:
   """Resizes the image to the specified size.
 
   Args: