org.apache.flink.runtime.checkpoint.TaskState to org.apache.flink.runtime.checkpoint.OperatorState