Skip to content

Commit 4a526bf

Browse files
committed
fix ssm data docstring examples
1 parent f2effc0 commit 4a526bf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

doubleml/data/ssm_data.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,9 @@ class DoubleMLSSMData(DoubleMLData):
6666
>>> from doubleml.irm.datasets import make_ssm_data
6767
>>> # initialization from pandas.DataFrame
6868
>>> df = make_ssm_data(return_type='DataFrame')
69-
>>> obj_dml_data_from_df = DoubleMLSSMData(df, 'y', 'd', 's')
69+
>>> obj_dml_data_from_df = DoubleMLSSMData(df, 'y', 'd', s_col='s')
7070
>>> # initialization from np.ndarray
71-
>>> (x, y, d, s) = make_ssm_data(return_type='array')
71+
>>> (x, y, d, _, s) = make_ssm_data(return_type='array')
7272
>>> obj_dml_data_from_array = DoubleMLSSMData.from_arrays(x, y, d, s=s)
7373
"""
7474

@@ -186,7 +186,7 @@ def from_arrays(
186186
--------
187187
>>> from doubleml import DoubleMLSSMData
188188
>>> from doubleml.irm.datasets import make_ssm_data
189-
>>> (x, y, d, s) = make_ssm_data(return_type='array')
189+
>>> (x, y, d, _, s) = make_ssm_data(return_type='array')
190190
>>> obj_dml_data_from_array = DoubleMLSSMData.from_arrays(x, y, d, s=s)
191191
"""
192192
# Prepare selection variable

0 commit comments

Comments
 (0)