set num_processes=1 in MAB test

Summary: Attempt to fix the test timing out on CircleCI Differential Revision: D33900623 fbshipit-source-id: ad8b7b2f0da95b109921490ecdb47ed77767800b
facebookresearch · Jan 31, 2022 · c34704e · c34704e
1 parent 09e36a9
commit c34704e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/reagent/test/mab/test_mab.py b/reagent/test/mab/test_mab.py
@@ -374,6 +374,7 @@ def test_multiple_evaluations_bandit_algo(self):
             max_steps=max_steps,
             algo_kwargs={"n_arms": 2},
             bandit_kwargs={"probs": torch.Tensor([0.3, 0.5])},
+            num_processes=1,
         )
 
         self.assertIsInstance(regret_trajectory, np.ndarray)