@Override public Schema outputSchema(final Schema input) { Env.setProperty(TopValueIndex.class, "vector.type", VectorUtils.typeOfVector(input)); return SchemaUtils.schemaOf(SchemaUtils.bagOf(SchemaUtils.primitiveOf(DataType.INTEGER))); }