Skip to content

Commit a07520a

Browse files
authored
minor modifciation (#425)
* minor modifciation * type
1 parent e1f0c88 commit a07520a

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

onnx_diagnostic/torch_models/validate.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2332,7 +2332,9 @@ def call_torch_export_custom(
23322332
"custom-fake",
23332333
"custom-tracing",
23342334
}
2335-
assert exporter in available, f"Unexpected value for exporter={exporter!r} in {available}"
2335+
assert (
2336+
exporter in available
2337+
), f"Unexpected value for exporter={exporter!r} in {sorted(available)}" # type: ignore[type-var]
23362338
assert "model" in data, f"model is missing from data: {sorted(data)}"
23372339
assert "inputs_export" in data, f"inputs_export is missing from data: {sorted(data)}"
23382340
assert ("-strict" not in exporter) or ("strict" not in exporter_options), (

0 commit comments

Comments
 (0)