fix issue with N not being used

drbenvincent · drbenvincent · commit e2902448f0d5 · 2025-07-19T08:02:52.000+01:00
diff --git a/causalpy/data/simulate_data.py b/causalpy/data/simulate_data.py
@@ -125,12 +125,12 @@ def generate_time_series_data(
         The intercept
 
     """
-    x = np.arange(0, 100, 1)
+    x = np.arange(0, N, 1)
     df = pd.DataFrame(
         {
             "temperature": np.sin(x * 0.5) + 1,
-            "linear": np.linspace(0, 1, 100),
-            "causal effect": 10 * gamma(10).pdf(np.arange(0, 100, 1) - treatment_time),
+            "linear": np.linspace(0, 1, N),
+            "causal effect": 10 * gamma(10).pdf(np.arange(0, N, 1) - treatment_time),
         }
     )
 

Original file line number	Diff line number	Diff line change
`@@ -125,12 +125,12 @@ def generate_time_series_data(`
`125`	`125`	`The intercept`
`126`	`126`
`127`	`127`	`"""`
`128`		`- x = np.arange(0, 100, 1)`
	`128`	`+ x = np.arange(0, N, 1)`
`129`	`129`	`df = pd.DataFrame(`
`130`	`130`	`{`
`131`	`131`	`"temperature": np.sin(x * 0.5) + 1,`
`132`		`- "linear": np.linspace(0, 1, 100),`
`133`		`- "causal effect": 10 * gamma(10).pdf(np.arange(0, 100, 1) - treatment_time),`
	`132`	`+ "linear": np.linspace(0, 1, N),`
	`133`	`+ "causal effect": 10 * gamma(10).pdf(np.arange(0, N, 1) - treatment_time),`
`134`	`134`	`}`
`135`	`135`	`)`
`136`	`136`