pytest "set_deterministic"

3c106aa4 · panning · 6b4739b5 · 3c106aa4
Commit 3c106aa4 authored Nov 14, 2022 by panning
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

test/test_functional_tensor.py test/test_functional_tensor.py +3 -1

No files found.
--- a/test/test_functional_tensor.py
+++ b/test/test_functional_tensor.py
@@ -1087,7 +1087,9 @@ def test_autocontrast(device, dtype):

 @pytest.mark.parametrize('device', cpu_and_gpu())
 def test_equalize(device):
-    torch.set_deterministic(False)
+    #pan
+    #torch.set_deterministic(False)
+    torch.use_deterministic_algorithms(False)
    check_functional_vs_PIL_vs_scripted(
        F.equalize,
        F_pil.equalize,