diff --git a/monai/utils/module.py b/monai/utils/module.py index ad070d06513..9b04308ea84 100644 --- a/monai/utils/module.py +++ b/monai/utils/module.py @@ -26,7 +26,7 @@ from pydoc import locate from re import match from types import FunctionType, ModuleType -from typing import Any, Iterable, cast, Dict, List +from typing import Any, Iterable, cast import torch @@ -60,7 +60,7 @@ "pytorch_after", ] -WRITER_PACKAGE_MAP: Dict[str, List[str]] = { +WRITER_PACKAGE_MAP: dict[str, list[str]] = { "png": ["pillow"], "jpg": ["pillow"], "jpeg": ["pillow"],