#pragma once

#include "grouped_units.h"
#include "sample.h"

#include <random>

namespace maps::wiki::assessment::sampler::impl {

GroupNameToUnits
pickBasicUnits(Mode mode, std::mt19937& rndGen, GroupNameToUnits&& groupNameToUnits);

std::vector<TId>
getOrCreateUnits(assessment::Gateway& assessmentGw, const GroupNameToUnits& groupNameToUnits);

} // namespace maps::wiki::assessment::sampler::impl
