in src/server/selection/banditData.ts [101:119]
function calculateMeanPerVariant(
samples: TestSample[],
test: BanditTestConfig,
): BanditVariantData[] {
const allVariantSamples = samples.flatMap((sample) => sample.variants);
return test.variantNames.map((variantName) => {
const variantSamples = allVariantSamples.filter(
(variantSample) => variantSample.variantName === variantName,
);
const mean = calculateOverallMeanForVariant(variantSamples);
return {
variantName,
mean,
};
});
}