From fc9be8f81f673c0b04a0381a544b78e5a2a0d0bb Mon Sep 17 00:00:00 2001 From: Jonas Knobloch Date: Tue, 24 Mar 2026 12:55:17 +0100 Subject: Refactor progress bar life cycle --- llm/perplexity.go | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) (limited to 'llm') diff --git a/llm/perplexity.go b/llm/perplexity.go index ab31889..5d9056e 100644 --- a/llm/perplexity.go +++ b/llm/perplexity.go @@ -1,7 +1,6 @@ package llm import ( - "context" "fmt" "sync" "time" @@ -47,16 +46,14 @@ func (e *Evaluator[R]) Run(data dataset.Reader, window, stride int) error { }() } - ctx, cancel := context.WithCancel(context.Background()) - - defer cancel() - pb := tui.NewProgressBar("Perplexity", 20, 0, time.Now()) - go pb.Watch(ctx, 1*time.Second, func() int { + pb.Start(1*time.Second, func() int { return int(e.completed.Load()) }) + defer pb.Close() + n := 0 for d := range data.Texts("text") { -- cgit v1.3.1