Merge pull request #56 from dailab/55-fix-wrong-typing-after-upgrade

55 fix wrong typing after upgrade

Author: EricKolibacz

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "maxi"
-version = "0.5.1"
+version = "0.5.2"
 description = "Model-Agnostic Explanation Library"
 authors = ["Tuan Anh, Le <[email protected]>"]
 license = "MIT"

src/maxi/lib/inference/inference_wrapper.py

@@ -44,9 +44,9 @@ def __call__(self, img: ndarray, *args, **kwargs) -> ndarray:
             quantizer method.
 
         Args:
-            img (NDArray[Int]): Image data in [width, height, channels]
+            img (ndarray[Int]): Image data in [width, height, channels]
 
         Returns:
-            NDArray[Float]: Array containing a valued class distribution, N entries for N classes. 
+            ndarray[Float]: Array containing a valued class distribution, N entries for N classes. 
         """
         return self.quantizer(self.inference_model(self.preprocess(img)))

src/maxi/lib/inference/quantizer/identity_quantizer.py

@@ -1,7 +1,4 @@
 """Identity Quantizer Method"""
-from typing import Any
-from nptyping import NDArray
-
 import numpy as np
 
 from .base_quantizer import BaseQuantizer
@@ -15,13 +12,13 @@ def __init__(self) -> None:
         """
         super().__init__()
 
-    def __call__(self, prediction: np.ndarray, *args, **kwargs) -> NDArray[Any]:
+    def __call__(self, prediction: np.ndarray, *args, **kwargs) -> np.ndarray:
         """Returns the resulting annotation as it is.
 
         Args:
             prediction (np.ndarray): Arbitrary prediction.
 
         Returns:
-            NDArray[Any]: The same arbitrary prediction from the input.
+            ndarray[Any]: The same arbitrary prediction from the input.
         """
         return super().__call__(prediction)