From 94d49f80252065cdc73ebc9bb78be78142a27ce3 Mon Sep 17 00:00:00 2001
From: Karla Saur <1703543+ksaur@users.noreply.github.com>
Date: Sun, 4 Aug 2024 21:06:00 -0700
Subject: [PATCH] mitigating onnx install issue (#789)

---
 .github/workflows/pythonapp.yml | 14 +++++++-------
 setup.py                        |  1 +
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml
index b184710f..445cb1b4 100644
--- a/.github/workflows/pythonapp.yml
+++ b/.github/workflows/pythonapp.yml
@@ -138,19 +138,19 @@ jobs:
         cd ~/work/hummingbird/tvm/python
         python3 -m pip install -e .
 
-    # We don't run pytest for macos py3.9 since we do coverage for that case.
+    # We don't run pytest for ubuntu py3.9 since we do coverage for that case.
     - name: Test with pytest
-      if: ${{ matrix.python-version != '3.9' || startsWith(matrix.os, 'macos') != true }}
+      if: ${{ matrix.python-version != '3.9' || startsWith(matrix.os, 'ubuntu') != true }}
       run: pytest
-    # Run and push coverage only for macos py3.9
-    - name: Coverage 3.9 macos
-      if: ${{ matrix.python-version == '3.9' && startsWith(matrix.os, 'macos') }}
+    # Run and push coverage only for ubuntu py3.9
+    - name: Coverage 3.9 ubuntu
+      if: ${{ matrix.python-version == '3.9' && startsWith(matrix.os, 'ubuntu') }}
       run: |
         coverage run -a -m pytest tests
         coverage xml
     - name: Upload coverage to Codecov
-      if: ${{ matrix.python-version == '3.9' && startsWith(matrix.os, 'macos') }}
-      uses: codecov/codecov-action@v3
+      if: ${{ matrix.python-version == '3.9' && startsWith(matrix.os, 'ubuntu') }}
+      uses: codecov/codecov-action@v4
       with:
         file: ./coverage.xml
         flags: unittests
diff --git a/setup.py b/setup.py
index dca5a7a5..f7b68645 100644
--- a/setup.py
+++ b/setup.py
@@ -22,6 +22,7 @@
 install_requires = [
     "numpy>=1.15",
     "onnxconverter-common>=1.6.0",
+    "onnx<=1.16.1",
     "scipy",
     "scikit-learn",
     "torch>1.7.0",