update for DoubleMLSSM data class

SvenKlaassen · SvenKlaassen · commit fa6edbbce5e2 · 2025-09-08T07:49:43.000+02:00
diff --git a/monte-cover/src/montecover/ssm/ssm_mar_ate.py b/monte-cover/src/montecover/ssm/ssm_mar_ate.py
@@ -119,5 +119,5 @@ def _generate_dml_data(self, dgp_params: Dict[str, Any]) -> dml.DoubleMLData:
             mar=True,
             return_type="DataFrame",
         )
-        dml_data = dml.DoubleMLData(data, "y", "d", s_col="s")
+        dml_data = dml.data.DoubleMLSSMData(data, "y", "d", s_col="s")
         return dml_data
diff --git a/monte-cover/src/montecover/ssm/ssm_nonig_ate.py b/monte-cover/src/montecover/ssm/ssm_nonig_ate.py
@@ -121,5 +121,5 @@ def _generate_dml_data(self, dgp_params: Dict[str, Any]) -> dml.DoubleMLData:
             mar=False,
             return_type="DataFrame",
         )
-        dml_data = dml.DoubleMLData(data, "y", "d", z_cols="z", s_col="s")
+        dml_data = dml.data.DoubleMLSSMData(data, "y", "d", z_cols="z", s_col="s")
         return dml_data

Original file line number	Diff line number	Diff line change
`@@ -119,5 +119,5 @@ def _generate_dml_data(self, dgp_params: Dict[str, Any]) -> dml.DoubleMLData:`
`119`	`119`	`mar=True,`
`120`	`120`	`return_type="DataFrame",`
`121`	`121`	`)`
`122`		`- dml_data = dml.DoubleMLData(data, "y", "d", s_col="s")`
	`122`	`+ dml_data = dml.data.DoubleMLSSMData(data, "y", "d", s_col="s")`
`123`	`123`	`return dml_data`
Original file line number	Diff line number	Diff line change
`@@ -121,5 +121,5 @@ def _generate_dml_data(self, dgp_params: Dict[str, Any]) -> dml.DoubleMLData:`
`121`	`121`	`mar=False,`
`122`	`122`	`return_type="DataFrame",`
`123`	`123`	`)`
`124`		`- dml_data = dml.DoubleMLData(data, "y", "d", z_cols="z", s_col="s")`
	`124`	`+ dml_data = dml.data.DoubleMLSSMData(data, "y", "d", z_cols="z", s_col="s")`
`125`	`125`	`return dml_data`