fix import issue

xadupre · xadupre · commit 2ea8a9497892 · 2025-11-18T11:44:53.000+01:00
diff --git a/CHANGELOGS.rst b/CHANGELOGS.rst
@@ -4,6 +4,7 @@ Change Logs
 0.8.3
 +++++
 
+* :pr:`310`: split patches into multiple files 
 * :pr:`308`: add option --save_ep to dump the exported program as well as torch input
 * :pr:`304`, :pr:`306`: improves side-by-side comparison, creates command line sbs
 
diff --git a/_unittests/ut_torch_export_patches/test_patch_rewriting.py b/_unittests/ut_torch_export_patches/test_patch_rewriting.py
@@ -1,6 +1,6 @@
 import unittest
 from onnx_diagnostic.ext_test_case import ExtTestCase
-from onnx_diagnostic.torch_export_patches.patches.patch_transformers import (
+from onnx_diagnostic.torch_export_patches.patches._patch_transformers_qwen2 import (
     rewrite_loop_for_square_mask,
 )
 from onnx_diagnostic.torch_export_patches.patch_module_helper import code_needing_rewriting

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`import unittest`
`2`	`2`	`from onnx_diagnostic.ext_test_case import ExtTestCase`
`3`		`-from onnx_diagnostic.torch_export_patches.patches.patch_transformers import (`
	`3`	`+from onnx_diagnostic.torch_export_patches.patches._patch_transformers_qwen2 import (`
`4`	`4`	`rewrite_loop_for_square_mask,`
`5`	`5`	`)`
`6`	`6`	`from onnx_diagnostic.torch_export_patches.patch_module_helper import code_needing_rewriting`