File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -253,7 +253,8 @@ def generate_regression_discontinuity_data(
253
253
>>> import pathlib
254
254
>>> from causalpy.data.simulate_data import generate_regression_discontinuity_data
255
255
>>> df = generate_regression_discontinuity_data(true_treatment_threshold=0.5)
256
- >>> df.to_csv(pathlib.Path.cwd() / 'regression_discontinuity.csv', index=False)
256
+ >>> df.to_csv(pathlib.Path.cwd() / 'regression_discontinuity.csv',
257
+ ... index=False) # doctest: +SKIP
257
258
"""
258
259
259
260
def is_treated (x ):
@@ -288,7 +289,8 @@ def generate_ancova_data(
288
289
... treatment_effect=2,
289
290
... sigma=1
290
291
... )
291
- >>> df.to_csv(pathlib.Path.cwd() / 'ancova_data.csv', index=False)
292
+ >>> df.to_csv(pathlib.Path.cwd() / 'ancova_data.csv',
293
+ ... index=False) # doctest: +SKIP
292
294
"""
293
295
group = np .random .choice (2 , size = N )
294
296
pre = np .random .normal (loc = pre_treatment_means [group ])
You can’t perform that action at this time.
0 commit comments