use deterministic torch for test_linear_transformation (pytorch#3099)

Quansight · Dec 4, 2020 · b30a49b · b30a49b
1 parent b24ed2e
commit b30a49b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_transforms_tensor.py b/test/test_transforms_tensor.py
@@ -452,6 +452,7 @@ def test_linear_transformation(self):
         # We skip some tests from _test_transform_vs_scripted_on_batch as
         # results for scripted and non-scripted transformations are not exactly the same
         torch.manual_seed(12)
+        torch.set_deterministic(True)
         transformed_batch = fn(batch_tensors)
         torch.manual_seed(12)
         s_transformed_batch = scripted_fn(batch_tensors)