support decollate for numpy scalars

fix linter

Signed-off-by: Arthur Dujardin <arthurdujardin.dev@gmail.com>

fix numpy decollate multi arrays

Signed-off-by: Arthur Dujardin <arthurdujardin.dev@gmail.com>

fix linter

Signed-off-by: Arthur Dujardin <arthurdujardin.dev@gmail.com>

fix numpy scalar support

Signed-off-by: Arthur Dujardin <arthurdujardin.dev@gmail.com>

Author: arthurdjn

monai/data/utils.py

@@ -621,10 +621,13 @@ def decollate_batch(batch, detach: bool = True, pad=True, fill_value=None):
     """
     if batch is None:
         return batch
-    if isinstance(batch, (float, int, str, bytes)) or (
-        type(batch).__module__ == "numpy" and not isinstance(batch, Iterable)
-    ):
+    if isinstance(batch, (float, int, str, bytes)):
         return batch
+    if type(batch).__module__ == "numpy":
+        if not isinstance(batch, Iterable):
+            return batch
+        if batch.ndim == 0:
+            return batch.item() if detach else batch
     if isinstance(batch, torch.Tensor):
         if detach:
             batch = batch.detach()