Update tests to utilize replication

JMGaljaard · Sep 6, 2022 · ecf4035 · ecf4035
1 parent 454c6c6
commit ecf4035
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/util/config/test_learning_config.py b/tests/util/config/test_learning_config.py
@@ -32,7 +32,8 @@ class FedLearningConfigTest(unittest.TestCase):
                                 scheduler_step_size=50,
                                 scheduler_gamma=0.5,
                                 min_lr=1e-10,
-                                optimizer=Optimizations.sgd)
+                                optimizer=Optimizations.sgd,
+                                replication=-1)
     def setUp(self):
         self.learning_params = FedLearningConfig.from_yaml(Path(TEST_PARAM_CONF_FEDERATED))