Skip to content

Commit efa6880

Browse files
committed
refactor patches
1 parent 8bc74a1 commit efa6880

File tree

4 files changed

+696
-546
lines changed

4 files changed

+696
-546
lines changed

_unittests/ut_torch_export_patches/test_patch_details.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,16 @@ def test_patch_details(self):
3232
rst = patch.format_diff(format="rst")
3333
self.assertIn("====", rst)
3434

35+
# second time to make every patch was removed
36+
with torch_export_patches(
37+
patch_transformers=True,
38+
verbose=10,
39+
patch_torch=True,
40+
patch_diffusers=True,
41+
patch_details=details,
42+
):
43+
pass
44+
3545
@requires_transformers("4.55")
3646
def test_patch_diff(self):
3747
from onnx_diagnostic.torch_export_patches.patches.patch_transformers import (

0 commit comments

Comments
 (0)