test(causal_effect): reduce test accuracy

causy-dev · Aug 4, 2024 · 480cb45 · 480cb45
1 parent f948ca5
commit 480cb45
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_effect_estimation.py b/tests/test_effect_estimation.py
@@ -75,22 +75,22 @@ def test_direct_effect_estimation(self):
             tst.graph.edge_value(tst.graph.nodes["X"], tst.graph.nodes["Y"])[
                 "direct_effect"
             ],
-            5,
-            1,
+            5.0,
+            0,
         )
         self.assertAlmostEqual(
             tst.graph.edge_value(tst.graph.nodes["Z"], tst.graph.nodes["Y"])[
                 "direct_effect"
             ],
-            6,
-            1,
+            6.0,
+            0,
         )
         self.assertAlmostEqual(
             tst.graph.edge_value(tst.graph.nodes["W"], tst.graph.nodes["Y"])[
                 "direct_effect"
             ],
-            2,
-            1,
+            2.0,
+            0,
         )
 
     def test_direct_effect_estimation_weird_graph(self):