From b36b9a014559eba76fdebc025d0dd59395903e5c Mon Sep 17 00:00:00 2001
From: tchaton <thomas@grid.ai>
Date: Fri, 27 Nov 2020 18:50:01 +0000
Subject: [PATCH] reduce parity test

---
 benchmarks/test_parity.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmarks/test_parity.py b/benchmarks/test_parity.py
index d026096a84..05973e19a5 100644
--- a/benchmarks/test_parity.py
+++ b/benchmarks/test_parity.py
@@ -12,7 +12,7 @@ from tests.base.models import ParityModuleMNIST, ParityModuleRNN
 # ParityModuleMNIST runs with num_workers=1
 @pytest.mark.parametrize('cls_model,max_diff', [
     (ParityModuleRNN, 0.05),
-    (ParityModuleMNIST, 0.22)
+    (ParityModuleMNIST, 0.18)
 ])
 @pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires GPU machine")
 def test_pytorch_parity(tmpdir, cls_model, max_diff):