Fix mypy and tests after updates

bhky · Dec 4, 2023 · 0d41095 · 0d41095
1 parent 4212e7f
commit 0d41095
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/targetran/_np_functional.py b/targetran/_np_functional.py
@@ -3,7 +3,7 @@
 """
 from typing import Tuple
 
-import cv2  # type: ignore
+import cv2
 import numpy as np
 
 from targetran._typing import (
@@ -80,7 +80,7 @@ def _np_boolean_mask(x: NDAnyArray, mask: NDBoolArray) -> NDAnyArray:
     """
     mask: boolean array
     """
-    return x[mask]  # type: ignore
+    return x[mask]
 
 
 def _np_gather_image(image: NDFloatArray, indices: NDIntArray) -> NDFloatArray:

diff --git a/targetran/np/_np.py b/targetran/np/_np.py
@@ -286,7 +286,7 @@ def _get_mats(
             )
         else:
             conditions = rand_fn() < probs
-            indices = np.arange(len(probs), dtype=np.int32)[conditions]
+            indices = np.arange(len(probs), dtype=np.int32)[conditions]  # type: ignore
 
         if len(indices) > 1:
             if self._keep_order:

diff --git a/tests/__init__.py b/tests/__init__.py