Skip to content

Commit 5da9c0b

Browse files
committed
Removing change to get_most_recent_transform
Signed-off-by: Eric Kerfoot <eric.kerfoot@kcl.ac.uk>
1 parent 000b901 commit 5da9c0b

2 files changed

Lines changed: 2 additions & 15 deletions

File tree

monai/transforms/inverse.py

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -336,25 +336,13 @@ def get_most_recent_transform(self, data, key: Hashable = None, check: bool = Tr
336336
else:
337337
raise ValueError(f"`data` should be either `MetaTensor` or dictionary, got {type(data)}.")
338338

339-
# Find the last transform whose name matches that of this class, this allows Invertd to ignore applied
340-
# operations added by transforms it is not trying to invert, ie. those added in postprocessing.
341-
idx = -1
342-
for i in reversed(range(len(all_transforms))):
343-
xform_name = all_transforms[i].get(TraceKeys.CLASS_NAME, "")
344-
if xform_name == self.__class__.__name__:
345-
idx = i # if nothing found, idx remains -1 so replicating previous behaviour
346-
break
347-
348339
if not all_transforms:
349340
raise ValueError(f"Item of type {type(data)} (key: {key}, pop: {pop}) has empty 'applied_operations'")
350341

351342
if check:
352-
if not (-len(all_transforms) <= idx < len(all_transforms)):
353-
raise IndexError(f"Index '{idx}' not valid for list of applied operations '{all_transforms}'")
354-
355-
self.check_transforms_match(all_transforms[idx])
343+
self.check_transforms_match(all_transforms[-1])
356344

357-
return all_transforms.pop(idx) if pop else all_transforms[idx]
345+
return all_transforms.pop(-1) if pop else all_transforms[-1]
358346

359347
def pop_transform(self, data, key: Hashable = None, check: bool = True):
360348
"""

tests/transforms/inverse/test_inverse_dict.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,6 @@ def setUp(self):
3939

4040
self.postprocessing = Compose(
4141
[
42-
Lambdad(self.pred, func=lambda x: x), # tests that added postprocess transforms don't confuse Invertd
4342
Invertd(self.pred, transform=self.preprocessing, orig_keys=self.key),
4443
]
4544
)

0 commit comments

Comments
 (0)