fix hypothesis timeout (#26280)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26280

ghstack-source-id: 90160270

Test Plan: testinprod

Differential Revision: D17396861

fbshipit-source-id: ee2348ffa7f6092e2c5647a42d0e17879dcfacd0
This commit is contained in:
Sebastian Messmer 2019-09-16 09:08:03 -07:00 committed by Facebook Github Bot
parent 38b2bc1451
commit c006356034

View File

@ -137,6 +137,7 @@ class TestQuantizedOps(TestCase):
self.assertEqual(qY, qY_hat, message="{} relu failed".format(name))
"""Tests the correctness of the scalar addition."""
@no_deadline
@given(A=hu.tensor(shapes=hu.array_shapes(1, 4, 1, 5),
elements=st.floats(-1e6, 1e6, allow_nan=False),
qparams=hu.qparams()),