fix constraint for pixdim in track_transform_meta() function

Signed-off-by: Wei_Chuan, Chiang <[email protected]> Co-authored-by: einsyang723 <[email protected]> Co-authored-by: IamTingTing <[email protected]>
Project-MONAI · Feb 19, 2025 · a34a656 · a34a656
1 parent b6dd460
commit a34a656
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/monai/transforms/inverse.py b/monai/transforms/inverse.py
@@ -197,8 +197,9 @@ def track_transform_meta(
                 else:
                     raise
             out_obj.meta[MetaKeys.AFFINE] = convert_to_tensor(affine, device=torch.device("cpu"), dtype=torch.float64)
-            spacing = affine_to_spacing(out_obj.meta[MetaKeys.AFFINE])
-            out_obj.meta[MetaKeys.PIXDIM][1 : 1 + len(spacing)] = spacing
+            if MetaKeys.PIXDIM in out_obj.meta:
+                spacing = affine_to_spacing(out_obj.meta[MetaKeys.AFFINE])
+                out_obj.meta[MetaKeys.PIXDIM][1 : 1 + len(spacing)] = spacing
 
         if not (get_track_meta() and transform_info and transform_info.get(TraceKeys.TRACING)):
             if isinstance(data, Mapping):