From 49d495488a38b509031dad58dc71546eac2d55c7 Mon Sep 17 00:00:00 2001 From: Arthur Dujardin Date: Tue, 3 Jun 2025 10:13:49 +0000 Subject: [PATCH] support decollate for numpy scalars fix linter Signed-off-by: Arthur Dujardin fix numpy decollate multi arrays Signed-off-by: Arthur Dujardin fix linter Signed-off-by: Arthur Dujardin fix numpy scalar support Signed-off-by: Arthur Dujardin minor refactoring for typing Signed-off-by: Arthur Dujardin convert scalar array to tensor Signed-off-by: Arthur Dujardin --- monai/data/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/monai/data/utils.py b/monai/data/utils.py index 14217e9103..ee3bf10bf6 100644 --- a/monai/data/utils.py +++ b/monai/data/utils.py @@ -625,6 +625,8 @@ def decollate_batch(batch, detach: bool = True, pad=True, fill_value=None): type(batch).__module__ == "numpy" and not isinstance(batch, Iterable) ): return batch + if isinstance(batch, np.ndarray) and batch.ndim == 0: + batch = torch.from_numpy(batch) if isinstance(batch, torch.Tensor): if detach: batch = batch.detach()