diff --git a/requirements/segmentation_test.txt b/requirements/segmentation_test.txt index f3428aef485..56fd7bc4c62 100644 --- a/requirements/segmentation_test.txt +++ b/requirements/segmentation_test.txt @@ -2,4 +2,4 @@ # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment scipy >1.0.0, <1.17.0 -monai ==1.4.0 +monai ==1.5.1 diff --git a/tests/unittests/segmentation/test_utils.py b/tests/unittests/segmentation/test_utils.py index 30a6da7d954..1b9d77a672b 100644 --- a/tests/unittests/segmentation/test_utils.py +++ b/tests/unittests/segmentation/test_utils.py @@ -233,7 +233,7 @@ def test_mask_edges(cases, spacing, crop, device): reference_res = monai_get_mask_edges(preds, target, spacing=spacing, crop=crop) for r1, r2 in zip(res, reference_res): - assert torch.allclose(r1.cpu().float(), torch.from_numpy(r2).float()) + assert torch.allclose(r1.cpu().float(), r2.cpu().float()) @pytest.mark.parametrize(