From b344910d1c4c3b3bfd455cb55a8958a893a94b62 Mon Sep 17 00:00:00 2001 From: brash6 Date: Thu, 14 Nov 2024 15:33:17 +0100 Subject: [PATCH] fix tests --- src/tests/estimation/get_estimation_results.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tests/estimation/get_estimation_results.py b/src/tests/estimation/get_estimation_results.py index 3c2f1bf..fe98338 100644 --- a/src/tests/estimation/get_estimation_results.py +++ b/src/tests/estimation/get_estimation_results.py @@ -37,7 +37,7 @@ def _transform_outputs(causal_effects): indirect_treated = causal_effects['indirect_effect_treated'] indirect_control = causal_effects['indirect_effect_control'] - return np.array(total, direct_treated, direct_control, indirect_treated, indirect_control, 0) + return [total, direct_treated, direct_control, indirect_treated, indirect_control, 0] def _get_estimation_results(x, t, m, y, estimator, config):