Fixed strategy equality bug

jmafoster1 · jmafoster1 · commit 50b2bf2d2d4a · 2024-09-18T11:52:41.000+01:00
diff --git a/causal_testing/estimation/ipcw_estimator.py b/causal_testing/estimation/ipcw_estimator.py
@@ -172,7 +172,7 @@ def preprocess_data(self):
             )
 
             strategy_followed = [
-                (t, var, individual.loc[individual["time"] == t, var].values[0])
+                [t, var, individual.loc[individual["time"] == t, var].values[0]]
                 for t, var, val in self.treatment_strategy
             ]
 

Original file line number	Diff line number	Diff line change
`@@ -172,7 +172,7 @@ def preprocess_data(self):`
`172`	`172`	`)`
`173`	`173`
`174`	`174`	`strategy_followed = [`
`175`		`- (t, var, individual.loc[individual["time"] == t, var].values[0])`
	`175`	`+ [t, var, individual.loc[individual["time"] == t, var].values[0]]`
`176`	`176`	`for t, var, val in self.treatment_strategy`
`177`	`177`	`]`
`178`	`178`