From 7ae0651a55ace3c5d1f651775c1160f6e0b72152 Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Sun, 7 Jun 2026 20:26:59 +0200
Subject: Add config to lesci experiment

---
 research/lesci/experiment.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'research/lesci/experiment.go')

diff --git a/research/lesci/experiment.go b/research/lesci/experiment.go
index df30f2c..e625301 100644
--- a/research/lesci/experiment.go
+++ b/research/lesci/experiment.go
@@ -21,9 +21,11 @@ type Experiment struct {
 	name           string
 	cutoff         int
 	window         int
+	config         Config
+	options        Options
 }
 
-func NewExperiment(model llm.Causal, tokenizer, counterfactual llm.Tokenizer, data dataset.Reader, name string, cutoff, window int) (*Experiment, error) {
+func NewExperiment(model llm.Causal, tokenizer, counterfactual llm.Tokenizer, data dataset.Reader, name string, cutoff, window int, cfg Config, opts Options) (*Experiment, error) {
 	e := &Experiment{
 		model:          model,
 		tokenizer:      tokenizer,
@@ -32,6 +34,8 @@ func NewExperiment(model llm.Causal, tokenizer, counterfactual llm.Tokenizer, da
 		name:           name,
 		cutoff:         cutoff,
 		window:         window,
+		config:         cfg,
+		options:        opts,
 	}
 
 	return e, nil
-- 
cgit v1.3.1