fix tests

judithabk6 · Nov 14, 2024 · b344910 · b344910
1 parent 9bff44b
commit b344910
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):