Improves rewriting

[xadupre]

No description provided.

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 62.60870% with 43 lines in your changes missing coverage. Please review.

Project coverage is 74.41%. Comparing base (54e8373) to head (c7f8b58).
Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
...nx_diagnostic/torch_export_patches/patch_module.py	35.29%	33 Missing ⚠️
...gnostic/torch_export_patches/onnx_export_errors.py	33.33%	4 Missing ⚠️
onnx_diagnostic/torch_models/test_helper.py	73.33%	4 Missing ⚠️
onnx_diagnostic/torch_export_patches/__init__.py	0.00%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #105      +/-   ##
==========================================
- Coverage   74.49%   74.41%   -0.09%     
==========================================
  Files          84       85       +1     
  Lines        8194     8269      +75     
==========================================
+ Hits         6104     6153      +49     
- Misses       2090     2116      +26     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.