Skip to content

Commit 03919f9

Browse files
committed
fix
1 parent e68412e commit 03919f9

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

onnx_diagnostic/export/api.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,9 @@ def to_onnx(
191191

192192
irfunctions = [
193193
ir.from_proto(
194-
plug.get_function_proto(*flatten_object((args, kwargs), drop_keys=True))
194+
plug.get_function_proto(
195+
target_opset, *flatten_object((args, kwargs), drop_keys=True)
196+
)
195197
)
196198
for plug in onnx_plugs
197199
]

0 commit comments

Comments
 (0)