fixup

Signed-off-by: Fabian Klopfer <[email protected]>

Author: SomeUserName1

monai/transforms/utility/dictionary.py

@@ -1479,7 +1479,10 @@ def __init__(self, keys: KeysCollection, name: str, allow_missing_keys: bool = F
         self.trans = TorchIO(name, *args, **kwargs)
 
     def __call__(self, data: Mapping[Hashable, NdarrayOrTensor]) -> dict[Hashable, NdarrayOrTensor]:
-        return self.trans(dict(data))
+        d = dict(data)
+        for key in self.key_iterator(d):
+            d[key] = self.trans(d[key])
+        return d
 
 
 class RandTorchIOd(MapTransform, RandomizableTrait):
@@ -1509,7 +1512,11 @@ def __init__(self, keys: KeysCollection, name: str, allow_missing_keys: bool = F
         self.trans = TorchIO(name, *args, **kwargs)
 
     def __call__(self, data: Mapping[Hashable, NdarrayOrTensor]) -> dict[Hashable, NdarrayOrTensor]:
-        return self.trans(dict(data))
+        d = dict(data)
+        for key in self.key_iterator(d):
+            d[key] = self.trans(d[key])
+        return d
+
 
 
 class MapLabelValued(MapTransform):