serializ

Author: thomaschow

spark/src/main/scala/ai/chronon/spark/submission/ChrononKryoRegistrator.scala

@@ -28,6 +28,7 @@ import org.apache.datasketches.cpc.CpcSketch
 import org.apache.datasketches.frequencies.ItemsSketch
 import org.apache.datasketches.memory.Memory
 import org.apache.spark.serializer.KryoRegistrator
+import ai.chronon.aggregator.windowing.SawtoothMutationAggregator
 
 import java.lang.invoke.SerializedLambda
 
@@ -219,6 +220,7 @@ class ChrononKryoRegistrator extends KryoRegistrator {
       kryo.register(Class.forName(name))
       kryo.register(Class.forName(s"[L$name;")) // represents array of a type to jvm
       kryo.register(classOf[SawtoothAggregator], new JavaSerializer)
+      kryo.register(classOf[SawtoothMutationAggregator], new JavaSerializer)
     } catch {
       case _: ClassNotFoundException => // do nothing
     }