@Override public Datum getPartialResult(FunctionContext ctx) { AvgContext avgCtx = (AvgContext) ctx; ArrayDatum part = new ArrayDatum(2); part.put(0, DatumFactory.createDouble(avgCtx.sum)); part.put(1, DatumFactory.createLong(avgCtx.count)); return part; }