@@ -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