fix mypy (#181)

pyronear · Apr 4, 2024 · e53e357 · e53e357
1 parent 3838aa3
commit e53e357
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyroengine/engine.py b/pyroengine/engine.py
@@ -290,7 +290,7 @@ def predict(self, frame: Image.Image, cam_id: Optional[str] = None) -> float:
         cam_key = cam_id or "-1"
         # Reduce image size to save bandwidth
         if isinstance(self.frame_size, tuple):
-            frame_resize = frame.resize(self.frame_size[::-1], Image.BILINEAR)
+            frame_resize = frame.resize(self.frame_size[::-1], getattr(Image, "BILINEAR"))
         else:
             frame_resize = frame