|
22 | 22 | from executorch.extension.pybindings.portable_lib import ( # @manual |
23 | 23 | _load_for_executorch_from_buffer, |
24 | 24 | ) |
25 | | -from hypothesis import given, settings, strategies as st |
26 | 25 | from torch.export import export |
27 | 26 |
|
28 | 27 |
|
@@ -65,7 +64,6 @@ def forward(self, *args): |
65 | 64 | .executorch_program |
66 | 65 | ) |
67 | 66 |
|
68 | | - @settings(deadline=500000) |
69 | 67 | def test_emit_lowered_backend_module_end_to_end(self): |
70 | 68 | class SinModule(torch.nn.Module): |
71 | 69 | def __init__(self): |
@@ -109,11 +107,7 @@ def forward(self, x): |
109 | 107 | torch.allclose(model_outputs[0], expected_res, atol=1e-03, rtol=1e-03) |
110 | 108 | ) |
111 | 109 |
|
112 | | - @given( |
113 | | - unlift=st.booleans(), # verify both lifted and unlifted graph |
114 | | - ) |
115 | | - @settings(deadline=500000) |
116 | | - def test_emit_lowered_backend_module(self, unlift): |
| 110 | + def test_emit_lowered_backend_module(self): |
117 | 111 | module_list = [ |
118 | 112 | models.Emformer(), |
119 | 113 | models.Repeat(), |
@@ -166,11 +160,7 @@ def test_emit_lowered_backend_module(self, unlift): |
166 | 160 | _ = lowered_model.buffer() |
167 | 161 | self.validate_lowered_module_program(program) |
168 | 162 |
|
169 | | - @given( |
170 | | - unlift=st.booleans(), # verify both lifted and unlifted graph |
171 | | - ) |
172 | | - @settings(deadline=500000) |
173 | | - def test_emit_nested_lowered_backend_module(self, unlift): |
| 163 | + def test_emit_nested_lowered_backend_module(self): |
174 | 164 | module_list = [ |
175 | 165 | models.Emformer(), |
176 | 166 | models.Repeat(), |
|
0 commit comments