Skip to content

Commit 3c6e2dc

Browse files
authored
added completed code for tracking (#603)
1 parent fa0748d commit 3c6e2dc

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

k8s-bench/eval.go

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,8 +149,18 @@ func runEvaluation(ctx context.Context, config EvalConfig) error {
149149
log = logFile
150150
}
151151

152+
start := time.Now()
153+
fmt.Printf("\033[36mWorker %d: Started %s for %s\033[0m\n", workerID, llmConfig.ID, job.taskID)
154+
152155
result := evaluateTask(ctx, config, job.taskID, job.task, llmConfig, log)
153156

157+
fmt.Printf("\033[32mWorker %d: Completed %s for %s in %s\033[0m\n",
158+
workerID,
159+
llmConfig.ID,
160+
job.taskID,
161+
time.Since(start).Round(time.Second),
162+
)
163+
154164
if taskOutputDir != "" {
155165
if err := writeToYAMLFile(filepath.Join(taskOutputDir, "results.yaml"), result); err != nil {
156166
errorsCh <- fmt.Errorf("writing results to file: %w", err)

0 commit comments

Comments
 (0)