From ab46556a0ac2f43d4aeed8f546362e15b39a0609 Mon Sep 17 00:00:00 2001
From: Alessandro Clerici <a.clericilorenzini@reply.it>
Date: Wed, 9 Jul 2025 15:40:27 +0000
Subject: [PATCH] more bearable pyproject.toml

---
 pyproject.toml | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index a094ed9..fefd567 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,5 +1,6 @@
 [build-system]
-requires = ["setuptools>=69.5"]
+# requires = ["setuptools>=69.5"]
+requires = ["setuptools"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]
@@ -13,20 +14,23 @@ readme = "README.md"
 requires-python = ">=3.10"
 classifiers = ["Programming Language :: Python :: 3"]
 dependencies = [
-    "accelerate==1.1.0",
+    # "accelerate==1.1.0",
+    "accelerate",
     "loguru",
     "requests",
-    "setuptools==69.5",
+    # "setuptools>=69.5",
+    "setuptools",
     "transformers",
     "tokenizers",
-    "flash-attn==2.5.8",
-    "bitsandbytes==0.43.1",
+    # "flash-attn==2.5.8",
+    "flash-attn",
+    # "bitsandbytes==0.43.1",
+    "bitsandbytes",
     "safetensors",
     "gpustat",
     "icecream",
     "pdf2image",
-    "numpy==1.26.4",
-    "torchvision",
+    "numpy",
     "jsonlines",
     "editdistance",
     "einops",
@@ -34,13 +38,17 @@ dependencies = [
     "peft",
     "timm",
     "sentencepiece",
-    "colpali-engine==0.3.1",
+    # "colpali-engine==0.3.1",
+    "colpali-engine",
     "easyocr",
     "qwen-vl-utils",
     "faiss-cpu",
     "word2number",
-    "datasets>=3.0.0",
+    # "datasets>=3.0.0",
     "python-dotenv",
+    # "torchvision==0.22.1+cu128",
+    "torchvision",
+    "datasets>=3.6.0",
 ]
 
 [tool.ruff]
@@ -50,4 +58,4 @@ target-version = "py310"
 select = ["E4", "E7", "E9", "F", "I"]
 
 [tool.ruff.lint.per-file-ignores]
-"__init__.py" = ["F401"]
\ No newline at end of file
+"__init__.py" = ["F401"]