Make sure ONNX ops are run with opset bigger than 7.

For some reason, for some ops, the opset version is set to the miniumum supported version. For certain ops, this is version 1. The ONNX runtime is printing a warning about any version less than 7. So, if the opset version is less than 7, opset_version.
ModECI · Mar 19, 2024 · f3e524d · f3e524d
1 parent 065fb63
commit f3e524d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/modeci_mdf/functions/onnx.py b/src/modeci_mdf/functions/onnx.py
@@ -152,6 +152,13 @@ def run_onnx_op(
     )
 
     model_def = op.to_onnx(inputs)
+
+    # For some reason, for some ops, the opset version is set to the miniumum supported version.
+    # For certain ops, this is version 1. The ONNX runtime is printing a warning about any version
+    # less than 7. So, if the opset version is less than 7, set to opset_version.
+    if model_def.opset_import[0].version < 7:
+        model_def.opset_import[0].version = opset_version
+
     return predict_with_onnxruntime(model_def, *input_vals)