class DatasetAffinityFunctionWrapper implements AffinityFunction {

    @Override public int partition(Object key) {
        return (Integer) key;
    }

    public  DatasetAffinityFunctionWrapper(AffinityFunction delegate);

}