docs: def get(self, class_name):
          """Get a transform in the chain from its name."""
          for transform in self.cpu_transforms + self.gpu_transforms:
              if transform.__class__.__name__ == class_name:
                  return transform