From ea59fdcaa2900c2572b40f1a91f06b9962685e12 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Mon, 13 Sep 2021 14:20:26 +0100 Subject: [PATCH] fixes #2939 Signed-off-by: Wenqi Li --- monai/_extensions/loader.py | 2 +- monai/data/grid_dataset.py | 2 +- monai/data/utils.py | 4 ++-- monai/transforms/inverse_batch_transform.py | 2 +- monai/utils/aliases.py | 4 ++-- tests/utils.py | 2 +- 6 files changed, 8 insertions(+), 8 deletions(-) diff --git a/monai/_extensions/loader.py b/monai/_extensions/loader.py index 5f77480ecc..6c68fe08c7 100644 --- a/monai/_extensions/loader.py +++ b/monai/_extensions/loader.py @@ -34,7 +34,7 @@ def timeout(time, message): except KeyboardInterrupt as e: if timer is not None and timer.is_alive(): raise e # interrupt from user? - raise TimeoutError(message) + raise TimeoutError(message) from e finally: if timer is not None: try: diff --git a/monai/data/grid_dataset.py b/monai/data/grid_dataset.py index 5b2a4d7abd..5c330f10e4 100644 --- a/monai/data/grid_dataset.py +++ b/monai/data/grid_dataset.py @@ -141,7 +141,7 @@ def __iter__(self): try: iter_end = len(self.dataset) # TODO: support iterable self.dataset except TypeError: - raise NotImplementedError("image dataset must implement `len()`.") + raise NotImplementedError("image dataset must implement `len()`.") from None if worker_info is not None: # split workload diff --git a/monai/data/utils.py b/monai/data/utils.py index aab23217dc..a5cb5057d4 100644 --- a/monai/data/utils.py +++ b/monai/data/utils.py @@ -283,7 +283,7 @@ def list_data_collate(batch: Sequence): + "`DataLoader` with `collate_fn=pad_list_data_collate` might solve this problem (check its " + "documentation)." ) - raise RuntimeError(re_str) + raise RuntimeError(re_str) from re except TypeError as re: re_str = str(re) if "numpy" in re_str and "Tensor" in re_str: @@ -294,7 +294,7 @@ def list_data_collate(batch: Sequence): + "creating your `DataLoader` with `collate_fn=pad_list_data_collate` might solve this problem " + "(check its documentation)." ) - raise TypeError(re_str) + raise TypeError(re_str) from re def decollate_batch(batch, detach: bool = True): diff --git a/monai/transforms/inverse_batch_transform.py b/monai/transforms/inverse_batch_transform.py index d9c6790840..b485e5bac4 100644 --- a/monai/transforms/inverse_batch_transform.py +++ b/monai/transforms/inverse_batch_transform.py @@ -99,7 +99,7 @@ def __call__(self, data: Dict[str, Any]) -> Any: re_str = str(re) if "equal size" in re_str: re_str += "\nMONAI hint: try creating `BatchInverseTransform` with `collate_fn=lambda x: x`." - raise RuntimeError(re_str) + raise RuntimeError(re_str) from re class Decollated(MapTransform): diff --git a/monai/utils/aliases.py b/monai/utils/aliases.py index 2b7b29eeb5..a08dab4f95 100644 --- a/monai/utils/aliases.py +++ b/monai/utils/aliases.py @@ -70,8 +70,8 @@ def resolve_name(name): try: mod = importlib.import_module(modname) obj = getattr(mod, declname, None) - except ModuleNotFoundError: - raise ValueError(f"Module {modname!r} not found.") + except ModuleNotFoundError as not_found_err: + raise ValueError(f"Module {modname!r} not found.") from not_found_err if obj is None: raise ValueError(f"Module {modname!r} does not have member {declname!r}.") diff --git a/tests/utils.py b/tests/utils.py index 1375cd2d72..3364085e13 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -74,7 +74,7 @@ def test_pretrained_networks(network, input_param, device): try: net = network(**input_param).to(device) except (URLError, HTTPError, ContentTooShortError) as e: - raise unittest.SkipTest(e) + raise unittest.SkipTest(e) from e return net