bug

Author: xadupre

CHANGELOGS.rst

@@ -4,7 +4,7 @@ Change Logs
 0.3.0
 +++++
 
-* :pr:`30`: adds command to test a model id
+* :pr:`30`: adds command to test a model id, validate the export
 * :pr:`29`: adds helpers to measure the memory peak and run benchmark
   on different processes
 * :pr:`28`: adds command line to print out the configuration for a model id,

_unittests/ut_helpers/test_torch_test_helper.py

@@ -62,6 +62,7 @@ def test_replace_string_by_dynamic(self):
         proc = replace_string_by_dynamic(example)
         sproc = (
             str(proc)
+            .replace(",min=None,max=None,_factory=True", "")
             .replace("_DimHint(type=<_DimHintType.DYNAMIC: 3>)", "DYN")
             .replace(" ", "")
             .replace("<_DimHint.DYNAMIC:3>", "DYN")

_unittests/ut_torch_models/test_test_helpers.py

@@ -42,6 +42,7 @@ def test_validate_model_export(self):
             verbose=10,
             exporter="export-nostrict",
             dump_folder="dump_test_validate_model_export",
+            patch=True,
         )
         self.assertIsInstance(summary, dict)
         self.assertIsInstance(data, dict)

onnx_diagnostic/torch_models/test_helper.py

@@ -21,6 +21,9 @@ def empty(value: Any) -> bool:
 def _ds_clean(v):
     return (
         str(v)
+        .replace(",min=None", "")
+        .replace(",max=None", "")
+        .replace(",_factory=True", "")
         .replace("<class 'onnx_diagnostic.torch_models.hghub.model_inputs.", "")
         .replace("'>", "")
         .replace("_DimHint(type=<_DimHintType.DYNAMIC: 3>)", "DYNAMIC")