How to write a case class for an enumeration column in Apache Spark Dataset?

Question

story1, 10, small
story2, 20, medium
sotry3, 3, small
story4, 50, xlarge

I want to convert my data to Dataset. I have a column nameType (small, medium, large, xlarge). Therefore, I do not know how to write a case class in this situation

case class Story(name:String, point: Int, storyType: ???)

Answer 1

Try

sealed trait StoryType
case object small extends StoryType
case object medium extends StoryType
case object large extends StoryType
case object xlarge extends StoryType