module.exports = {
  name: "train",
  scheduleSeconds: 30,
  async produceJobs(queue) {
    await queue.add("train", { steps: 200 }, {
      attempts: 2,
      backoff: { type: "fixed", delay: 2000 },
      removeOnComplete: 200,
      removeOnFail: 200
    });
  },
  async process(job, log) {
    const { trainSteps } = require("../lib/training");
    const t0 = Date.now();
    const steps = Number(job.data?.steps || 200);
    const meta = trainSteps(steps);
    log.info({ steps, ms: Date.now() - t0, loss: meta.loss.toFixed(4), step: meta.step }, "train tick");
  }
};