✨ update library dependencies/ GPU&CPU installation

Author: HamedBabaei

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "OntoLearner"
-version = "0.0.0"  # placeholder, will be replaced automatically
+version = "0.0.0"
 description = "OntoLearner: A Modular Python Library for Ontology Learning with LLMs."
 authors = ["Hamed Babaei Giglou <[email protected]>", "Andrei C. Aioanei <[email protected]>"]
 license = "MIT License"
@@ -10,25 +10,35 @@ repository = "https://github.com/sciknoworg/OntoLearner/"
 include = ["images/logo.png"]
 
 [tool.poetry.dependencies]
-python=">=3.10,<3.14.0"
-rdflib="7.1.1"
-networkx ="3.2.1"
-seaborn="*"
-numpy="*"
-pandas="*"
-openpyxl="*"
-matplotlib="*"
-tqdm="*"
-pydantic="2.11.3"
-pathlib="1.0.1"
-dspy="2.6.14"
-python-dotenv="*"
-huggingface-hub="0.34.4"
-transformers="4.56.0"
-torch="2.8.0"
-sentence-transformers="5.1.0"
-scikit-learn="1.6.1"
-bitsandbytes="0.45.1"
+python = ">=3.10,<3.14.0"
+
+# Core deps
+rdflib = "7.1.1"
+networkx = "3.2.1"
+numpy = "*"
+pandas = "*"
+openpyxl = "*"
+tqdm = "*"
+pydantic = "2.11.3"
+pathlib = "1.0.1"
+python-dotenv = "*"
+matplotlib = "*"
+seaborn = "*"
+scikit-learn = "1.6.1"
+huggingface-hub = "0.34.4"
+
+# LLM stack (CPU by default)
+torch = { version = "^2.8.0", source = "pytorch-cpu" }
+transformers = "^4.56.0"
+sentence-transformers = "^5.1.0"
+dspy = "^2.6.14"
+
+[tool.poetry.extras]
+# GPU-only extras
+gpu = [
+    "torch==^2.8.0",        # GPU wheel (auto CUDA)
+    "bitsandbytes=^0.45.1"  # allow minor/patch updates
+]
 
 [tool.poetry.dev-dependencies]
 ruff = "*"
@@ -47,3 +57,8 @@ attr = "ontolearner.__version__"
 [build-system]
 requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.4.0"]
 build-backend = "poetry_dynamic_versioning.backend"
+
+[[tool.poetry.source]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+secondary = true

setup.py

@@ -26,13 +26,12 @@
         "networkx==3.2.1",
         "pydantic==2.11.3",
         "pathlib==1.0.1",
-        "dspy==2.6.14",
-        "huggingface-hub==0.34.4",
-        "transformers==4.56.0",
-        "torch==2.8.0",
-        "sentence-transformers==5.1.0",
-        "scikit-learn==1.6.1",
-        "bitsandbytes==0.45.1",
+        "dspy>=2.6.14,<3.0.0",
+        "huggingface-hub>=0.34.4,<1.0.0",
+        "transformers>=4.56.0,<5.0.0",
+        "sentence-transformers>=5.1.0,<6.0.0",
+        "scikit-learn>=1.6.1,<2.0.0",
+        "bitsandbytes>=0.45.1,<1.0.0"
     ],
     classifiers=[
         "Development Status :: 5 - Production/Stable",