|
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