fix sensitivity_plot benchmark_dict value for cf_d

DoubleML · Aug 5, 2024 · 0aa7c49 · 0aa7c49
1 parent dc5d347
commit 0aa7c49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doubleml/double_ml.py b/doubleml/double_ml.py
@@ -1685,7 +1685,7 @@ def sensitivity_plot(self, idx_treatment=0, value='theta', include_scenario=True
             benchmark_values = np.full(shape=(n_benchmarks,), fill_value=np.nan)
             for benchmark_idx in range(len(benchmarks['name'])):
                 sens_dict_bench = self._calc_sensitivity_analysis(cf_y=benchmarks['cf_y'][benchmark_idx],
-                                                                  cf_d=benchmarks['cf_y'][benchmark_idx],
+                                                                  cf_d=benchmarks['cf_d'][benchmark_idx],
                                                                   rho=self.sensitivity_params['input']['rho'],
                                                                   level=self.sensitivity_params['input']['level'])
                 benchmark_values[benchmark_idx] = sens_dict_bench[value][bound][idx_treatment]