diff --git a/tests/test_mednext.py b/tests/test_mednext.py index e5f74118c3..cdafb8660b 100644 --- a/tests/test_mednext.py +++ b/tests/test_mednext.py @@ -18,7 +18,6 @@ from monai.networks import eval_mode from monai.networks.nets import MedNeXt -from tests.utils import SkipIfBeforePyTorchVersion, test_script_save device = "cuda" if torch.cuda.is_available() else "cpu"