OnnxRuntimeException in spark

[sansar-choinyambuu]

We are loading an onnx model in scala spark and doing some inference. The spark job is running on GCP DataProc cluster. Most times the inference is successful. However we get an occasional spark job failure with the following error:

22/05/31 15:55:14 INFO com.microsoft.azure.synapse.ml.onnx.ONNXModel: metrics/ {"uid":"ONNXModel_ccf3f81e5884","className":"class com.microsoft.azure.synapse.ml.onnx.ONNXModel","method":"transform","buildVersion":"0.9.4"}
22/05/31 15:55:14 INFO com.microsoft.azure.synapse.ml.onnx.ONNXModel: metrics/ {"uid":"ONNXModel_ccf3f81e5884","className":"class com.microsoft.azure.synapse.ml.onnx.ONNXModel","method":"transform","buildVersion":"0.9.4"}
22/05/31 15:55:14 INFO com.microsoft.azure.synapse.ml.stages.FixedMiniBatchTransformer: metrics/ {"uid":"FixedMiniBatchTransformer_be680faac5c5","className":"class com.microsoft.azure.synapse.ml.stages.FixedMiniBatchTransformer","method":"transform","buildVersion":"0.9.4"}
22/05/31 15:55:14 INFO com.microsoft.azure.synapse.ml.stages.FixedMiniBatchTransformer: metrics/ {"uid":"FixedMiniBatchTransformer_be680faac5c5","className":"class com.microsoft.azure.synapse.ml.stages.FixedMiniBatchTransformer","method":"transform","buildVersion":"0.9.4"}
22/05/31 15:55:14 INFO com.microsoft.azure.synapse.ml.onnx.ONNXModel: metrics/ {"uid":"ONNXModel_ccf3f81e5884","className":"class com.microsoft.azure.synapse.ml.onnx.ONNXModel","method":"transform","buildVersion":"0.9.4"}

terminate called after throwing an instance of 'onnxruntime::OnnxRuntimeException'
  what():  /onnxruntime_src/onnxruntime/core/session/ort_env.cc:98 static void OrtEnv::Release(OrtEnv*) env_ptr == p_instance_ was false. 

The error occurs statistically, we cannot reliably reproduce the issue. Does anybody have some pointers which direction we should look?
We are using sbt dependancy com.microsoft.azure.synapseml 0.9.4