Skip to content

Commit 9ebe46f

Browse files
add CuCIM type hints, docstring fix, and skip_if_quick for tests
Signed-off-by: Nikolas Schmitz <nikolas.schmitz@rwth-aachen.de>
1 parent ecece0d commit 9ebe46f

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

monai/data/wsi_reader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -886,7 +886,7 @@ def get_wsi_at_mpp(
886886
closest_lvl = closest_lvl - 1
887887
closest_lvl_wsi = self._resize_to_mpp_res(wsi, closest_lvl, mpp_list, mpp)
888888

889-
wsi_arr = cp.asnumpy(closest_lvl_wsi)
889+
wsi_arr: np.ndarray = cp.asnumpy(closest_lvl_wsi)
890890

891891
# Ensure channel dimension exists for grayscale
892892
if wsi_arr.ndim < 3:

tests/utils/enums/test_wsireader.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from monai.transforms import Compose, LoadImaged, ToTensord
2828
from monai.utils import first, optional_import
2929
from monai.utils.enums import PostFix, WSIPatchKeys
30-
from tests.test_utils import assert_allclose, download_url_or_skip_test, skip_if_no_cuda, testing_data_config
30+
from tests.test_utils import assert_allclose, download_url_or_skip_test, skip_if_no_cuda, skip_if_quick, testing_data_config
3131

3232
cucim, has_cucim = optional_import("cucim")
3333
has_cucim = has_cucim and hasattr(cucim, "CuImage")
@@ -480,6 +480,7 @@ def test_read_whole_image(self, file_path, level, expected_shape):
480480
assert_allclose(meta[WSIPatchKeys.SIZE], expected_shape[1:], type_test=False)
481481
assert_allclose(meta[WSIPatchKeys.LOCATION], (0, 0), type_test=False)
482482

483+
@skip_if_quick
483484
@parameterized.expand(
484485
[
485486
TEST_CASE_SVS_MPP_1,

0 commit comments

Comments
 (0)