From 385ce1f903959ea20c0a4c98cb3f66b6197c058e Mon Sep 17 00:00:00 2001 From: Jakub Kaczmarzyk Date: Tue, 9 Jul 2024 13:55:31 -0400 Subject: [PATCH] update float and int types for slide coords and probs Use np.floating and np.integer. --- wsinfer/modellib/run_inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wsinfer/modellib/run_inference.py b/wsinfer/modellib/run_inference.py index 7ba8aa5..0a0926e 100644 --- a/wsinfer/modellib/run_inference.py +++ b/wsinfer/modellib/run_inference.py @@ -171,8 +171,8 @@ def run_inference( # Store the coordinates and model probabiltiies of each patch in this slide. # This lets us know where the probabiltiies map to in the slide. - slide_coords: list[npt.NDArray[np.int_]] = [] - slide_probs: list[npt.NDArray[np.float_]] = [] + slide_coords: list[npt.NDArray[np.integer]] = [] + slide_probs: list[npt.NDArray[np.floating]] = [] for batch_imgs, batch_coords in tqdm.tqdm(loader): assert batch_imgs.shape[0] == batch_coords.shape[0], "length mismatch" with torch.no_grad():