guh

mosaicml · Jun 12, 2024 · c4da63a · c4da63a
1 parent 329df18
commit c4da63a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/composer/algorithms/utils/augmentation_primitives.py b/composer/algorithms/utils/augmentation_primitives.py
@@ -155,7 +155,7 @@ def rotate(pil_img: Image.Image, level: float):
     degrees = _int_parameter(_sample_level(level), 30)
     if np.random.uniform() > 0.5:
         degrees = -degrees
-    return pil_img.rotate(degrees, resample=Image.BILINEAR)
+    return pil_img.rotate(degrees, resample=Image.Resampling.BILINEAR)
 
 
 def solarize(pil_img: Image.Image, level: float):