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 8f2a295..17a7759 100644 --- a/llm/evaluator.go +++ b/llm/evaluator.go @@ -40,7 +40,7 @@ func (e *Evaluator[R]) Results() chan R { return e.results } -func (e *Evaluator[R]) RunAndCollect(data dataset.Reader, window, stride int, callback func(R) error) error { +func (e *Evaluator[R]) RunAndCollect(title string, data dataset.Reader, window, stride int, callback func(R) error) error { var wg sync.WaitGroup var collectErr error @@ -57,7 +57,7 @@ func (e *Evaluator[R]) RunAndCollect(data dataset.Reader, window, stride int, ca } }() - runErr := e.Run(data, window, stride) + runErr := e.Run(title, data, window, stride) wg.Wait() |