set num_processes=1 in MAB test

alexnikulkov · facebook-github-bot · commit c34704e77e5f · 2022-01-31T13:19:14.000-08:00
Summary: Attempt to fix the test timing out on CircleCI

Differential Revision: D33900623

fbshipit-source-id: ad8b7b2f0da95b109921490ecdb47ed77767800b
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)

Original file line number	Diff line number	Diff line change
`@@ -374,6 +374,7 @@ def test_multiple_evaluations_bandit_algo(self):`
`374`	`374`	`max_steps=max_steps,`
`375`	`375`	`algo_kwargs={"n_arms": 2},`
`376`	`376`	`bandit_kwargs={"probs": torch.Tensor([0.3, 0.5])},`
	`377`	`+ num_processes=1,`
`377`	`378`	`)`
`378`	`379`
`379`	`380`	`self.assertIsInstance(regret_trajectory, np.ndarray)`