fixes tests

Signed-off-by: Wenqi Li <wenqil@nvidia.com>

Author: wyli

monai/data/meta_obj.py

@@ -126,6 +126,8 @@ def copy_meta_from(self, input_objs, copy_attr=True, keys=None):
         return self with the updated ``__dict__``.
         """
         first_meta = input_objs if isinstance(input_objs, MetaObj) else first(input_objs, default=self)
+        if not hasattr(first_meta, "__dict__"):
+            return self
         first_meta = first_meta.__dict__
         keys = first_meta.keys() if keys is None else keys
         if not copy_attr:

monai/transforms/croppad/array.py

@@ -43,9 +43,10 @@
     map_classes_to_indices,
     weighted_patch_samples,
 )
+from monai.utils import ImageMetaKey as Key
 from monai.utils import (
-    Method,
     LazyAttr,
+    Method,
     PytorchPadMode,
     TraceKeys,
     TransformBackends,
@@ -58,7 +59,6 @@
     fall_back_tuple,
     look_up_option,
     pytorch_after,
-    ImageMetaKey as Key,
 )
 
 __all__ = [

tests/test_traceable_transform.py

@@ -13,16 +13,18 @@
 
 import unittest
 
+import torch
+
 from monai.transforms.inverse import TraceableTransform
 
 
 class _TraceTest(TraceableTransform):
     def __call__(self, data):
-        self.push_transform(data)
+        self.push_transform(data, "image")
         return data
 
     def pop(self, data):
-        self.pop_transform(data)
+        self.pop_transform(data, "image")
         return data
 
 
@@ -34,21 +36,11 @@ def test_default(self):
 
         data = {"image": "test"}
         data = a(data)  # adds to the stack
-        self.assertTrue(isinstance(data[expected_key], list))
-        self.assertEqual(data[expected_key][0]["class"], "_TraceTest")
+        self.assertEqual(data["image"], "test")
 
+        data = {"image": torch.tensor(1.0)}
         data = a(data)  # adds to the stack
-        self.assertEqual(len(data[expected_key]), 2)
-        self.assertEqual(data[expected_key][-1]["class"], "_TraceTest")
-
-        with self.assertRaises(IndexError):
-            a.pop({"test": "test"})  # no stack in the data
-        data = a.pop(data)
-        data = a.pop(data)
-        self.assertEqual(data[expected_key], [])
-
-        with self.assertRaises(IndexError):  # no more items
-            a.pop(data)
+        self.assertEqual(data["image"].applied_operations[0]["class"], "_TraceTest")
 
 
 if __name__ == "__main__":