From f765e2603b14186574ec86ff70a1767adfec867d Mon Sep 17 00:00:00 2001
From: Robert Sachunsky <sachunsky@informatik.uni-leipzig.de>
Date: Wed, 4 Dec 2024 15:57:13 +0000
Subject: [PATCH] move Torch to optional dependencies (to avoid clash with TF
 over CuDNN)

---
 pyproject.toml   | 4 ++++
 requirements.txt | 4 ----
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index b056cb7..61d488a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,6 +25,10 @@ classifiers = [
     "Topic :: Scientific/Engineering :: Image Processing",
 ]
 
+[project.optional-dependencies]
+OCR = ["torch <= 2.0.1", "transformers <= 4.30.2"]
+plotting = ["matplotlib"]
+
 [project.scripts]
 eynollah = "eynollah.cli:main"
 ocrd-eynollah-segment = "eynollah.ocrd_cli:main"
diff --git a/requirements.txt b/requirements.txt
index 02450aa..d72df29 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,8 +4,4 @@ numpy <1.24.0
 scikit-learn >= 0.23.2
 tensorflow < 2.13
 imutils >= 0.5.3
-matplotlib
-setuptools >= 50
-transformers <= 4.30.2
-torch <= 2.0.1
 numba <= 0.58.1