Skip to content

Commit

Permalink
Add:Support for ONNX versions 1.8.*, 1.9.*, 1.10.* + Python3.9 (#179)
Browse files Browse the repository at this point in the history
  • Loading branch information
rahul-tuli authored Aug 26, 2021
1 parent 6420aed commit 9ba7f42
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@
# File regexes for binaries to include in package_data
binary_regexes = ["*/*.so", "*/*.so.*", "*.bin", "*/*.bin"]

_deps = ["numpy>=1.16.3", "onnx>=1.5.0,<1.8.0", "requests>=2.0.0", "tqdm>=4.0.0"]
_deps = ["numpy>=1.16.3", "onnx>=1.5.0,<=1.10.1", "requests>=2.0.0", "tqdm>=4.0.0"]
_nm_deps = [f"{'sparsezoo' if is_release else 'sparsezoo-nightly'}~={version_base}"]
_dev_deps = [
"beautifulsoup4==4.9.3",
Expand Down Expand Up @@ -112,7 +112,7 @@ def _check_supported_system(self):

def _check_supported_python_version(self):
supported_major = 3
supported_minor = [6, 7, 8]
supported_minor = [6, 7, 8, 9]

if (
sys.version_info[0] != supported_major
Expand Down

0 comments on commit 9ba7f42

Please sign in to comment.