Fix tests to accommodate new PyTorch SobolEngine

Summary: Updates test to fix failures caused by pytorch/pytorch#49710 Removes the comparison against specifc sample values to be more robust in the future (but retains the distributional tests) Reviewed By: danielrjiang Differential Revision: D26207258 fbshipit-source-id: 3838339d6674518385ee6cf37faa7f52dc3de9dc
Balandat · Feb 3, 2021 · e2d8820 · e2d8820
1 parent 908ec57
commit e2d8820
Show file tree

Hide file tree

Showing 3 changed files with 65 additions and 208 deletions.
diff --git a/test/optim/test_initializers.py b/test/optim/test_initializers.py
@@ -159,7 +159,7 @@ def test_gen_batch_initial_conditions(self):
                             self.assertEqual(raw_samps.shape, expected_raw_samps_shape)
 
     def test_gen_batch_initial_conditions_highdim(self):
-        d = 120
+        d = 2200  # 2200 * 10 (q) > 21201 (sobol max dim)
         bounds = torch.stack([torch.zeros(d), torch.ones(d)])
         for dtype in (torch.float, torch.double):
             bounds = bounds.to(device=self.device, dtype=dtype)