Add token buffer config

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()