| 654 | return d |
| 655 | |
| 656 | def inverse(self, data: Mapping[Hashable, Any]) -> dict[Hashable, Any]: |
| 657 | d = dict(data) |
| 658 | for key in self.key_iterator(d): |
| 659 | transform = self.get_most_recent_transform(d, key) |
| 660 | # Create inverse transform |
| 661 | fwd_indices = np.array(transform[TraceKeys.EXTRA_INFO]["indices"]) |
| 662 | inv_indices = np.argsort(fwd_indices) |
| 663 | inverse_transform = Transpose(inv_indices.tolist()) |
| 664 | # Apply inverse |
| 665 | d[key] = inverse_transform(d[key]) |
| 666 | # Remove the applied transform |
| 667 | self.pop_transform(d, key) |
| 668 | return d |
| 669 | |
| 670 | |
| 671 | class DeleteItemsd(MapTransform): |