increase deadline for test (#513)

Summary: Pull Request resolved: #513 Reviewed By: ashkan-software Differential Revision: D39636639 fbshipit-source-id: 115e3f946cceccbe368d98574e591c58db045d74
pytorch · Sep 26, 2022 · 5e632cd · 5e632cd
1 parent 60cd993
commit 5e632cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/opacus/tests/grad_samples/conv2d_test.py b/opacus/tests/grad_samples/conv2d_test.py
@@ -40,7 +40,7 @@ class Conv2d_test(GradSampleHooks_test):
         dilation=st.integers(1, 3),
         groups=st.integers(1, 16),
     )
-    @settings(deadline=10000)
+    @settings(deadline=30000)
     def test_conv2d(
         self,
         N: int,
@@ -117,7 +117,7 @@ def test_conv2d(
         dilation_h=st.integers(1, 3),
         dilation_w=st.integers(1, 3),
     )
-    @settings(deadline=10000)
+    @settings(deadline=30000)
     def test_unfold2d(
         self,
         B: int,

diff --git a/opacus/tests/grad_samples/conv3d_test.py b/opacus/tests/grad_samples/conv3d_test.py
@@ -37,7 +37,7 @@ class Conv3d_test(GradSampleHooks_test):
         dilation=st.sampled_from([1, (1, 2, 2)]),
         groups=st.integers(1, 16),
     )
-    @settings(deadline=10000)
+    @settings(deadline=30000)
     def test_conv3d(
         self,
         N: int,