From 7f21e76175cdb47c430b9c5a031a09686be94fb6 Mon Sep 17 00:00:00 2001
From: Felix Dittrich <felixdittrich92@gmail.com>
Date: Thu, 10 Oct 2024 19:02:18 +0200
Subject: [PATCH] [Build] update minor version & update torch to >= 2.0 (#1747)

---
 .conda/meta.yaml                           | 2 +-
 README.md                                  | 2 +-
 docs/source/getting_started/installing.rst | 2 +-
 pyproject.toml                             | 8 ++++----
 setup.py                                   | 2 +-
 5 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/.conda/meta.yaml b/.conda/meta.yaml
index 7feb3a1bf9..f377481f16 100644
--- a/.conda/meta.yaml
+++ b/.conda/meta.yaml
@@ -1,7 +1,7 @@
 {% set pyproject = load_file_data('../pyproject.toml', from_recipe_dir=True) %}
 {% set project = pyproject.get('project') %}
 {% set urls = pyproject.get('project', {}).get('urls') %}
-{% set version = environ.get('BUILD_VERSION', '0.9.1a0') %}
+{% set version = environ.get('BUILD_VERSION', '0.10.0a0') %}
 
 package:
   name: {{ project.get('name') }}
diff --git a/README.md b/README.md
index 2fc92971ff..d57228fda6 100644
--- a/README.md
+++ b/README.md
@@ -161,7 +161,7 @@ pip install "python-doctr[torch,viz,html,contib]"
 For MacBooks with M1 chip, you will need some additional packages or specific versions:
 
 - TensorFlow 2: [metal plugin](https://developer.apple.com/metal/tensorflow-plugin/)
-- PyTorch: [version >= 1.12.0](https://pytorch.org/get-started/locally/#start-locally)
+- PyTorch: [version >= 2.0.0](https://pytorch.org/get-started/locally/#start-locally)
 
 ### Developer mode
 
diff --git a/docs/source/getting_started/installing.rst b/docs/source/getting_started/installing.rst
index 46d4177b30..e764e734a7 100644
--- a/docs/source/getting_started/installing.rst
+++ b/docs/source/getting_started/installing.rst
@@ -17,7 +17,7 @@ Whichever OS you are running, you will need to install at least TensorFlow or Py
 For MacBooks with M1 chip, you will need some additional packages or specific versions:
 
 * `TensorFlow 2 Metal Plugin <https://developer.apple.com/metal/tensorflow-plugin/>`_
-* `PyTorch >= 1.12.0 <https://pytorch.org/get-started/locally/#start-locally>`_
+* `PyTorch >= 2.0.0 <https://pytorch.org/get-started/locally/#start-locally>`_
 
 Via Python Package
 ==================
diff --git a/pyproject.toml b/pyproject.toml
index 9745f8a7c4..613eb512e2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -58,8 +58,8 @@ tf = [
     "tf2onnx>=1.16.0,<2.0.0",  # cf. https://github.com/onnx/tensorflow-onnx/releases/tag/v1.16.0
 ]
 torch = [
-    "torch>=1.12.0,<3.0.0",
-    "torchvision>=0.13.0",
+    "torch>=2.0.0,<3.0.0",
+    "torchvision>=0.15.0",
     "onnx>=1.12.0,<3.0.0",
 ]
 html = [
@@ -101,8 +101,8 @@ dev = [
     "tf-keras>=2.15.0,<3.0.0",  # Keep keras 2 compatibility
     "tf2onnx>=1.16.0,<2.0.0",  # cf. https://github.com/onnx/tensorflow-onnx/releases/tag/v1.16.0
     # PyTorch
-    "torch>=1.12.0,<3.0.0",
-    "torchvision>=0.13.0",
+    "torch>=2.0.0,<3.0.0",
+    "torchvision>=0.15.0",
     "onnx>=1.12.0,<3.0.0",
     # Extras
     "weasyprint>=55.0",
diff --git a/setup.py b/setup.py
index f45f3f157d..13fd4515e3 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
 from setuptools import setup
 
 PKG_NAME = "python-doctr"
-VERSION = os.getenv("BUILD_VERSION", "0.9.1a0")
+VERSION = os.getenv("BUILD_VERSION", "0.10.0a0")
 
 
 if __name__ == "__main__":