Как написать класс case для столбца перечисления в Apache Spark Dataset?

Question

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

Я хочу преобразовать свои данные в Dataset. У меня есть имя столбца nameType (малый, средний, большой, xlarge). Так что я не знаю, как написать мой случай класс в этой ситуации

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

Answer 1

Попробуйте

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