From e1e4eab2e1a682d633125f3c71cc09c3e95b30d3 Mon Sep 17 00:00:00 2001 From: Jonas Knobloch Date: Sat, 27 Jun 2026 19:08:39 +0200 Subject: Add token buffer config --- llm/evaluator.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'llm/evaluator.go') diff --git a/llm/evaluator.go b/llm/evaluator.go index 8aa4a32..69c4202 100644 --- a/llm/evaluator.go +++ b/llm/evaluator.go @@ -45,7 +45,7 @@ func (e *Evaluator[R]) Results() chan R { return e.results } -func (e *Evaluator[R]) RunAndCollect(title string, data dataset.Reader, window, stride int, callback func(R) error) error { +func (e *Evaluator[R]) RunAndCollect(title string, data dataset.Reader, cfg TokenBufferConfig, callback func(R) error) error { var wg sync.WaitGroup var collectErr error @@ -62,7 +62,7 @@ func (e *Evaluator[R]) RunAndCollect(title string, data dataset.Reader, window, } }() - runErr := e.Run(title, data, window, stride) + runErr := e.Run(title, data, cfg) wg.Wait() -- cgit v1.3.1