Prevent onnx 1.16.2 from being deployed. (#208)

onnx · Aug 1, 2024 · aa4b398 · aa4b398
1 parent ab54002
commit aa4b398
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -31,7 +31,9 @@
     ],
     install_requires=[
         "invoke>=2.0.0",
-        "onnx>=1.11.0",
+        # 1.16.2 causes a crash:
+        #   ImportError: DLL load failed while importing onnx_cpp2py_export
+        "onnx>=1.11.0,<1.16.2",
         "onnxmltools==1.10.0",
         "onnxruntime >=1.10.1",
         "torch>=1.12.1",
@@ -45,9 +47,6 @@
         "pandas>=1.5.3",
         "fasteners",
         "GitPython>=3.1.40",
-        # Necessary until upstream packages account for the breaking
-        # change to numpy
-        "numpy<2.0.0",
         "psutil",
     ],
     classifiers=[],