From a47c401f6e3f886c2059caacf69a1452fcf905df Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Xavier=20Dupr=C3=A9?= Date: Tue, 24 Feb 2026 22:33:09 +0100 Subject: [PATCH 1/2] minor modifciation --- onnx_diagnostic/torch_models/validate.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/onnx_diagnostic/torch_models/validate.py b/onnx_diagnostic/torch_models/validate.py index 92649cf1..47bac073 100644 --- a/onnx_diagnostic/torch_models/validate.py +++ b/onnx_diagnostic/torch_models/validate.py @@ -2332,7 +2332,9 @@ def call_torch_export_custom( "custom-fake", "custom-tracing", } - assert exporter in available, f"Unexpected value for exporter={exporter!r} in {available}" + assert ( + exporter in available + ), f"Unexpected value for exporter={exporter!r} in {sorted(available)}" assert "model" in data, f"model is missing from data: {sorted(data)}" assert "inputs_export" in data, f"inputs_export is missing from data: {sorted(data)}" assert ("-strict" not in exporter) or ("strict" not in exporter_options), ( From 8444a5279ec5c4fedcf1f720a8f05796681ce0d5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Xavier=20Dupr=C3=A9?= Date: Tue, 24 Feb 2026 22:37:35 +0100 Subject: [PATCH 2/2] type --- onnx_diagnostic/torch_models/validate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnx_diagnostic/torch_models/validate.py b/onnx_diagnostic/torch_models/validate.py index 47bac073..ed164558 100644 --- a/onnx_diagnostic/torch_models/validate.py +++ b/onnx_diagnostic/torch_models/validate.py @@ -2334,7 +2334,7 @@ def call_torch_export_custom( } assert ( exporter in available - ), f"Unexpected value for exporter={exporter!r} in {sorted(available)}" + ), f"Unexpected value for exporter={exporter!r} in {sorted(available)}" # type: ignore[type-var] assert "model" in data, f"model is missing from data: {sorted(data)}" assert "inputs_export" in data, f"inputs_export is missing from data: {sorted(data)}" assert ("-strict" not in exporter) or ("strict" not in exporter_options), (