def get_cur_rot(self) -> torch.Tensor:
        """
        Return the underlying rotation in its current form

        Returns:
            The stored rotation
        """
        if self._rot_mats is not None:
            return self._rot_mats
        elif self._quats is not None:
            return self._quats
        else:
            raise ValueError("Both rotations are None")

        Return the underlying rotation in its current form

        Returns:
            The stored rotation
        
dpkd/transformers/src/transformers/models/esm/openfold_utils/rigid_utils.py