Skip to content

Commit 1aabdae

Browse files
committed
fix image_only flag issue
Signed-off-by: Wei_Chuan, Chiang <[email protected]>
1 parent c5ebb0d commit 1aabdae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

monai/transforms/spatial/dictionary.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -522,8 +522,8 @@ def __call__(self, data: Mapping[Hashable, torch.Tensor], lazy: bool | None = No
522522
output_spatial_shape=output_shape_k if should_match else None,
523523
lazy=lazy_,
524524
)
525-
if isinstance(d[key], MetaTensor) and "filename_or_obj" in d[key].meta:
526-
if is_supported_format(d[key].meta["filename_or_obj"], ["nii", "nii.gz"]):
525+
if isinstance(d[key], MetaTensor) and f"{key}_meta_dict" in d:
526+
if 'filename_or_obj' in d[key].meta and is_supported_format(d[key].meta['filename_or_obj'], ["nii", "nii.gz"]):
527527
d = transforms.sync_meta_info(key, d)
528528
if output_shape_k is None:
529529
output_shape_k = d[key].peek_pending_shape() if isinstance(d[key], MetaTensor) else d[key].shape[1:]

0 commit comments

Comments
 (0)