Fix seed in test_ModelListGP_fixed_noise (#2774)

saitcakmak · facebook-github-bot · commit 1dcd1158d191 · 2025-03-17T10:28:09.000-07:00
Summary: Pull Request resolved: #2774 As titled. This fails on GPU without the seed. Reviewed By: dme65 Differential Revision: D71323337 fbshipit-source-id: b2be9e8a79b130225fa3b4520350f16a86c9188a
diff --git a/test/models/test_model_list_gp_regression.py b/test/models/test_model_list_gp_regression.py
@@ -261,6 +261,7 @@ def test_ModelListGP(self) -> None:
                 )
 
     def test_ModelListGP_fixed_noise(self) -> None:
+        torch.manual_seed(0)
         for dtype, outcome_transform in itertools.product(
             (torch.float, torch.double), ("None", "Standardize")
         ):

Original file line number	Diff line number	Diff line change
`@@ -261,6 +261,7 @@ def test_ModelListGP(self) -> None:`
`261`	`261`	`)`
`262`	`262`
`263`	`263`	`def test_ModelListGP_fixed_noise(self) -> None:`
	`264`	`+ torch.manual_seed(0)`
`264`	`265`	`for dtype, outcome_transform in itertools.product(`
`265`	`266`	`(torch.float, torch.double), ("None", "Standardize")`
`266`	`267`	`):`