50 lines
1.7 KiB
JavaScript
50 lines
1.7 KiB
JavaScript
|
// Copyright (C) CVAT.ai Corporation
|
|||
|
|
//
|
|||
|
|
// SPDX-License-Identifier: MIT
|
|||
|
|
import { randomIntBetween, randomItem } from 'https://jslib.k6.io/k6-utils/1.4.0/index.js';
|
|||
|
|
import APIJobs from '../api/jobs.js';
|
|||
|
|
|
|||
|
|
const stages = ['annotation', 'validation', 'acceptance'];
|
|||
|
|
const states = ['new', 'in progress', 'completed', 'rejected'];
|
|||
|
|
// const types = ['annotation', 'ground_truth', 'consensus_replica'];
|
|||
|
|
const types = ['ground_truth'];
|
|||
|
|
const frameMethods = ['random_uniform', 'random_per_job'];
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* Utility: Create a random job payload
|
|||
|
|
* This builds a valid random request body according to the CVAT jobs API schema.
|
|||
|
|
*
|
|||
|
|
* @param {number} taskId - ID of an existing task
|
|||
|
|
* @param {number|null} assigneeId - ID of the user to assign the job to (or null)
|
|||
|
|
* @returns {Object} Randomized job payload
|
|||
|
|
*/
|
|||
|
|
export function createRandomJob(authToken, taskId, totalFrames, assigneeId = null) {
|
|||
|
|
const stage = randomItem(stages);
|
|||
|
|
const state = randomItem(states);
|
|||
|
|
const type = randomItem(types);
|
|||
|
|
const frameSelection = randomItem(frameMethods);
|
|||
|
|
|
|||
|
|
const payload = {
|
|||
|
|
assignee: assigneeId,
|
|||
|
|
stage,
|
|||
|
|
state,
|
|||
|
|
type,
|
|||
|
|
task_id: taskId,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// Add frame selection–specific fields
|
|||
|
|
if (frameSelection === 'random_uniform') {
|
|||
|
|
payload.frame_selection_method = 'random_uniform';
|
|||
|
|
payload.frame_count = 1;
|
|||
|
|
payload.random_seed = randomIntBetween(0, 10000);
|
|||
|
|
} else if (frameSelection === 'random_per_job') {
|
|||
|
|
payload.frame_selection_method = 'random_per_job';
|
|||
|
|
payload.frames_per_job_count = 1;
|
|||
|
|
payload.random_seed = randomIntBetween(0, 10000);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return APIJobs.createJob(authToken, payload);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export default { createRandomJob };
|