Fix/relax numpy version constraint

README.md

@@ -1,10 +1,10 @@
 ![promptolution](https://github.com/user-attachments/assets/84c050bd-61a1-4f2e-bc4e-874d9b4a69af)
 
-![Coverage](https://img.shields.io/badge/Coverage-87%25-green)
+![Coverage](https://img.shields.io/badge/Coverage-89%25-green)
 [![CI](https://github.com/finitearth/promptolution/actions/workflows/ci.yml/badge.svg?branch=main)](https://github.com/finitearth/promptolution/actions/workflows/ci.yml)
 [![Docs](https://github.com/finitearth/promptolution/actions/workflows/docs.yml/badge.svg?branch=main)](https://github.com/finitearth/promptolution/actions/workflows/docs.yml)
 ![Code Style](https://img.shields.io/badge/Code%20Style-black-black)
-![Python Versions](https://img.shields.io/badge/Python%20Versions-≥3.9-blue)
+![Python Versions](https://img.shields.io/badge/Python%20Versions-≥3.10-blue)
 
 
 

docs/release-notes/v2.0.1.md

@@ -0,0 +1,7 @@
+## Release v2.0.1
+### What's changed
+- updated python requirement to >=3.10 (as 3.9 will lose support after October 2025)
+- fixed numpy version constraints (thanks to @asalaria-cisco)
+- make dependencies groups extras optional
+
+**Full Changelog**: [here](https://github.com/finitearth/promptolution/compare/v2.0.0...v2.0.1)

mkdocs.yml

@@ -47,6 +47,7 @@ nav:
   - Home: index.md
   - Release Notes:
     - Overview: release-notes.md
+    - v2.0.1: release-notes/v2.0.1.md
     - v2.0.0: release-notes/v2.0.0.md
     - v1.4.0: release-notes/v1.4.0.md
     - v1.3.2: release-notes/v1.3.2.md

pyproject.toml

@@ -1,25 +1,40 @@
 [tool.poetry]
 name = "promptolution"
-version = "2.0.0"
+version = "2.0.1"
 description = "A framework for prompt optimization and a zoo of prompt optimization algorithms."
 authors = ["Tom Zehle, Moritz Schlager, Timo Heiß"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = "^3.9"
-numpy = "^1.26.0"
+python = "^3.10"
+numpy = ">=1.26.0, <3.0.0"
 pandas = "^2.2.2"
 tqdm = "^4.66.5"
 scikit-learn = "^1.5.2"
 fastparquet = "^2024.11.0"
+openai = {version = "^1.0.0", optional = true}
+requests = {version = "^2.31.0", optional = true}
+vllm = {version = "^0.8.3", optional = true}
+transformers = {version = "^4.48.0", optional = true}
 
+[tool.poetry.extras]
+api = ["openai", "requests"]
+vllm = ["vllm"]
+transformers = ["transformers"]
+
+[tool.poetry.group.api]
+optional = true
 [tool.poetry.group.api.dependencies]
 openai = "^1.0.0"
 requests = "^2.31.0"
 
+[tool.poetry.group.vllm]
+optional = true
 [tool.poetry.group.vllm.dependencies]
 vllm = "^0.8.3"
 
+[tool.poetry.group.transformers]
+optional = true
 [tool.poetry.group.transformers.dependencies]
 transformers = "^4.48.0"