augmentTrainingSamples function exported ✓ 100.0%

export function augmentTrainingSamples(
  config: AugmentationConfig = DEFAULT_AUGMENTATION,
): TrainingSample[] {
  const augmented: TrainingSample[] = [];
  const strategyFns: Record<string, (s: string) => string> = {
    shuffle: augmentShuffle,
    drop: (s: string) => augmentDrop(s, config.dropRate ?? 0.2),
    typo: augmentTypo,
  };

  for (const sample of TRAINING_SAMPLES) {
    for (let i = 0; i < config.multiplier; i++) {
      // Pick a random strategy
      const strategy =
        config.strategies[Math.floor(Math.random() * config.strategies.length)];
      const fn = strategyFns[strategy];
      if (fn) {
        const signalText = flattenStructuredSignals(sample.signals);
        augmented.push({
          signals: fromFlatSignals(fn(signalText)),
          category: sample.category,
          type: sample.type,
          source: "augmented",
          domain: sample.domain,
          difficulty: sample.difficulty,
          language: sample.language,
          domFeatures: sample.domFeatures,
        });
      }
    }
  }

  return augmented;
}