File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
src/components/Experimental/FineTuning Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff 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' ;
You can’t perform that action at this time.
0 commit comments