def load_checkpoint(self, *args: Any, **kwargs: Any) -> dict[str, Any]:
        """Uses the base ``checkpoint_io`` to load the checkpoint."""
        assert self.checkpoint_io is not None
        return self.checkpoint_io.load_checkpoint(*args, **kwargs)
Uses the base ``checkpoint_io`` to load the checkpoint.
src/lightning/pytorch/plugins/io/wrapper.py