amend

pytorch · Feb 5, 2024 · 2ea60a9 · 2ea60a9
1 parent 3bb5ecb
commit 2ea60a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_utils.py b/test/test_utils.py
@@ -363,8 +363,8 @@ class MockGym:
 
 @pytest.mark.parametrize("device", get_default_devices())
 def test_rng_decorator(device):
-    torch.manual_seed(10)
     with torch.device(device):
+        torch.manual_seed(10)
         s0a = torch.randn(3)
         with _rng_decorator(0):
             torch.randn(3)