getting rid of the bug of repeated calls (#7)

owkin · Jan 23, 2024 · 6e14b2d · 6e14b2d
1 parent d0efbad
commit 6e14b2d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fedeca/fedeca_core.py b/fedeca/fedeca_core.py
@@ -1,4 +1,5 @@
 """Federate causal inference on distributed data."""
+import copy
 import logging
 import sys
 import time
@@ -181,7 +182,8 @@ def __init__(
         self.l1_ratio = l1_ratio
         self.initial_step_size = initial_step_size
         self.learning_rate_strategy = learning_rate_strategy
-        self.num_rounds_list = num_rounds_list
+        # Careful about mutable default args
+        self.num_rounds_list = copy.deepcopy(num_rounds_list)
         self.timeout = timeout
         self.sleep_time = sleep_time
         self.damping_factor_nr = damping_factor_nr