From af6b77672cbaf19ea0a13d5aee4e0fba9060c954 Mon Sep 17 00:00:00 2001
From: Guillaume Alleon <guillaume.alleon@gmail.com>
Date: Wed, 8 Jan 2025 19:22:44 +0100
Subject: [PATCH] Relaxing dependencies to work with hf smolagents

---
 pyproject.toml | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index bf6737126..b864c0ed5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,12 +34,12 @@ classifiers = [
 
 dependencies = [
     "GPUtil",
-    "numpy<2.0.0",
-    "peft>=0.11.0,<0.12.0",
-    "pillow>=9.2.0,<11.0.0",
+    "numpy",
+    "peft>=0.11.0",
+    "pillow>=9.2.0",
     "requests",
     "torch>=2.2.0",
-    "transformers>=4.46.1,<4.47.0",
+    "transformers>=4.46.1",
 ]
 
 [project.optional-dependencies]
@@ -65,6 +65,11 @@ all = ["colpali-engine[dev]", "colpali-engine[train]"]
 [project.urls]
 homepage = "https://github.com/illuin-tech/colpali"
 
+[dependency-groups]
+dev = [
+    "pytest>=8.3.4",
+]
+
 [tool.pytest.ini_options]
 filterwarnings = ["ignore::Warning"]
 markers = ["slow: marks test as slow"]