Skip to content

Commit 874027f

Browse files
committed
Exclude vLLM jobs from the fine-tune listing
Signed-off-by: Brent Salisbury <[email protected]>
1 parent a6ebb41 commit 874027f

File tree

1 file changed

+7
-3
lines changed
  • src/components/Experimental/FineTuning

1 file changed

+7
-3
lines changed

src/components/Experimental/FineTuning/index.tsx

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,16 +77,20 @@ const FineTuning: React.FC = () => {
7777
const logsIntervals = useRef<{ [jobId: string]: NodeJS.Timeout }>({});
7878

7979
const mapJobType = (job: Job) => {
80-
let jobType: 'generate' | 'train' | 'pipeline' | 'model-serve';
80+
let jobType: 'generate' | 'train' | 'pipeline' | 'model-serve' | 'vllm-run';
81+
8182
if (job.job_id.startsWith('g-')) {
8283
jobType = 'generate';
8384
} else if (job.job_id.startsWith('p-')) {
8485
jobType = 'pipeline';
8586
} else if (job.job_id.startsWith('ml-')) {
8687
jobType = 'model-serve';
88+
} else if (job.job_id.startsWith('v-')) {
89+
jobType = 'vllm-run'; // New categorization for 'v-' jobs
8790
} else {
8891
jobType = 'train';
8992
}
93+
9094
return { ...job, type: jobType };
9195
};
9296

@@ -190,8 +194,8 @@ const FineTuning: React.FC = () => {
190194
};
191195

192196
const filteredJobs = jobs.filter((job) => {
193-
if (job.job_id.startsWith('ml-')) {
194-
return false; // Exclude model serve jobs from the dashboard list
197+
if (job.job_id.startsWith('ml-') || job.job_id.startsWith('v-')) {
198+
return false; // Exclude model serve and vllm prefixed jobs from the dashboard list
195199
}
196200
if (selectedStatus === 'successful') return job.status === 'finished';
197201
if (selectedStatus === 'pending') return job.status === 'running';

0 commit comments

Comments
 (0)