increase to 0.8.0 and temp pin onnx (#1365)

mindee · Nov 1, 2023 · bd5b995 · bd5b995
1 parent aadb96a
commit bd5b995
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -58,6 +58,7 @@ dependencies = [
 tf = [
     "tensorflow>=2.11.0,<3.0.0",  # cf. https://github.com/mindee/doctr/pull/1182
     "tf2onnx>=1.15.1,<2.0.0",  # cf.https://github.com/onnx/tensorflow-onnx/releases/tag/v1.15.1
+    "onnx>=1.12.0,!=1.15.0,<3.0.0",  # TODO: remove this line once tf2onnx supports onnx>=1.15.0
 ]
 torch = [
     "torch>=1.12.0,<3.0.0",
@@ -97,7 +98,9 @@ dev = [
     # PyTorch
     "torch>=1.12.0,<3.0.0",
     "torchvision>=0.13.0",
-    "onnx>=1.12.0,<3.0.0",
+    # TODO: remove the constraint once tf2onnx supports onnx>=1.15.0
+    # TODO: cf. https://github.com/onnx/tensorflow-onnx/pull/2252
+    "onnx>=1.12.0,!=1.15.0,<3.0.0",
     # Testing
     "pytest>=5.3.2",
     "coverage[toml]>=4.5.4",

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 from setuptools import setup
 
 PKG_NAME = "python-doctr"
-VERSION = os.getenv("BUILD_VERSION", "0.7.1a0")
+VERSION = os.getenv("BUILD_VERSION", "0.8.0a0")
 
 
 if __name__ == "__main__":