fix typeerrror

pyronear · Apr 22, 2024 · 93c1950 · 93c1950
1 parent 6e85489
commit 93c1950
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyroengine/engine.py b/pyroengine/engine.py
@@ -17,6 +17,7 @@
 
 import cv2  # type: ignore[import-untyped]
 import numpy as np
+from numpy.typing import NDArray
 from PIL import Image
 from pyroclient import client
 from requests.exceptions import ConnectionError
@@ -267,7 +268,7 @@ def _update_states(self, frame: Image.Image, preds: np.ndarray, cam_key: str) ->
 
         return conf
 
-    def predict(self, frame: Image.Image, cam_id: Optional[str] = None) -> np.ndarray[Any, Any]:
+    def predict(self, frame: Image.Image, cam_id: Optional[str] = None) -> np.ndarray:
         """Computes the confidence that the image contains wildfire cues
 
         Args: