Skip to content

Commit 9ee7aa7

Browse files
committed
val
1 parent 0c37bc3 commit 9ee7aa7

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

_doc/cmds/validate.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,10 +77,10 @@ the same model.
7777

7878
.. code-block::
7979
80-
python -m onnx_diagnostic validate -m arnir0/Tiny-LLM --run -v 1 --export exporter-nostrict -o dump_models --patch
80+
python -m onnx_diagnostic validate -m arnir0/Tiny-LLM --run -v 1 --export export-nostrict -o dump_models --patch
8181
8282
.. runpython::
8383

8484
from onnx_diagnostic._command_lines_parser import main
8585

86-
main("validate -m arnir0/Tiny-LLM --run -v 1 --export exporter-nostrict -o dump_models --patch".split())
86+
main("validate -m arnir0/Tiny-LLM --run -v 1 --export export-nostrict -o dump_models --patch".split())

onnx_diagnostic/torch_models/test_helper.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def validate_model(
168168
assert hash_inputs == string_type(data["inputs"], with_shape=True), (
169169
f"The model execution did modified the inputs:\n"
170170
f"before: {hash_inputs}\n"
171-
f" after: {string_type(data["inputs"], with_shape=True)}"
171+
f" after: {string_type(data['inputs'], with_shape=True)}"
172172
)
173173

174174
if exporter:
@@ -220,7 +220,7 @@ def validate_model(
220220
), (
221221
f"The model execution did modified the inputs:\n"
222222
f"before: {hash_inputs}\n"
223-
f" after: {string_type(data["inputs_export"], with_shape=True)}"
223+
f" after: {string_type(data['inputs_export'], with_shape=True)}"
224224
)
225225

226226
# data is modified inplace
@@ -425,6 +425,6 @@ def call_torch_export_export(
425425
assert hash_inputs == string_type(data["inputs_export"], with_shape=True), (
426426
f"The exported model execution did modified the inputs:\n"
427427
f"before: {hash_inputs}\n"
428-
f" after: {string_type(data["inputs_export"], with_shape=True)}"
428+
f" after: {string_type(data['inputs_export'], with_shape=True)}"
429429
)
430430
return summary, data

0 commit comments

Comments
 (0)