diff options
Diffstat (limited to 'llm/evaluator.go')
| -rw-r--r-- | llm/evaluator.go | 4 |
1 files changed, 2 insertions, 2 deletions
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() |