Mark it `private[spark]` or provide doc? Also, please add return type (my bad).