Always import npt

Signed-off-by: Eric Kerfoot <[email protected]>

Author: ericspod

monai/metrics/average_precision.py

@@ -12,12 +12,10 @@
 from __future__ import annotations
 
 import warnings
-from typing import TYPE_CHECKING, cast
+from typing import cast
 
 import numpy as np
-
-if TYPE_CHECKING:
-    import numpy.typing as npt
+import numpy.typing as npt
 
 import torch
 

monai/metrics/rocauc.py

@@ -12,12 +12,10 @@
 from __future__ import annotations
 
 import warnings
-from typing import TYPE_CHECKING, cast
+from typing import cast
 
 import numpy as np
-
-if TYPE_CHECKING:
-    import numpy.typing as npt
+import numpy.typing as npt
 
 import torch