From cc6b1510e8fb350a58400f3ba4faf24f2eb9f768 Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjunsuresh1987@gmail.com>
Date: Thu, 20 Jun 2024 18:29:04 +0530
Subject: [PATCH] Added onnx dependency for 3dunet, fixes #78

---
 script/app-mlperf-inference-nvidia/_cm.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/script/app-mlperf-inference-nvidia/_cm.yaml b/script/app-mlperf-inference-nvidia/_cm.yaml
index 9820921f6..e6c38b5d9 100644
--- a/script/app-mlperf-inference-nvidia/_cm.yaml
+++ b/script/app-mlperf-inference-nvidia/_cm.yaml
@@ -416,6 +416,8 @@ variations:
     - tags: get,generic-python-lib,_package.nibabel
     - tags: get,generic-python-lib,_pandas
       version_max: "1.5.3"
+    - tags: get,generic-python-lib,_package.onnx
+      version: 1.13.1
 
   3d-unet-99:
     group: model