tests passing at 10% threshold

owkin · Mar 8, 2024 · 2ca1810 · 2ca1810
1 parent 2b378de
commit 2ca1810
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fedeca/tests/test_fl_vs_pooled.py b/fedeca/tests/test_fl_vs_pooled.py
@@ -77,7 +77,7 @@ def setUpClass(
             treated_col=cls._treated_col,
             duration_col=cls._duration_col,
             event_col=cls._event_col,
-            num_rounds_list=[50, 50],
+            num_rounds_list=[20, 20],
         )
 
         if variance_method == "bootstrap":
@@ -142,7 +142,7 @@ class TestBtstFedECAEnd2End(TestFedECAEnd2End):
     """BtstIPTW tests class."""
 
     @classmethod
-    def setUpClass(cls, seed=42, n_bootstrap=2):
+    def setUpClass(cls, seed=42, n_bootstrap=200):
         """Use parent class setup with robust=True."""
         bootstrap_seeds = []
         cls.n_bootstrap = n_bootstrap