diff --git a/poetry.lock b/poetry.lock
index a6947f9..190a568 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -5324,4 +5324,4 @@ cffi = ["cffi (>=1.11)"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.10,<4"
-content-hash = "9ebe1cf42bac38d4d10da3c02c97e9a607916d91e0a5e70bffd840f7105002bb"
+content-hash = "664b5ea951e268ebefd6637f1658919397ab350d82dc44c849a4f5b28b04a8d8"
diff --git a/pyproject.toml b/pyproject.toml
index 3c0566f..8d99317 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,5 +1,5 @@
 [build-system]
-requires = ["poetry-core>=2.0.0,<3.0.0"]
+requires = ["poetry-core>=2.0.0"]
 build-backend = "poetry.core.masonry.api"
 
 
@@ -18,18 +18,39 @@ classifiers = [
 ]
 requires-python = ">=3.10,<4"
 dependencies = [
-    "click (>=8.1.7,<9.0.0)",
-    "datasets (>=3.0.1,<4.0.0)",
+    "click>=8.1.7",
+    "datasets>=3.0.1",
     "tokenizers>=0.20.0",
-    "ray[default] (>=2.34.0,<3.0.0)",
-    "pyyaml (>=6.0.2,<7.0.0)",
-    "omegaconf (>=2.3.0,<3.0.0)",
-    "torch (>=2.4.1,<3.0.0)",
-    "requests (>=2.32.3,<3.0.0)",
+    "ray[default]>=2.34.0",
+    "pyyaml>=6.0.2",
+    "omegaconf>=2.3.0",
+    "torch>=2.4.1",
+    "requests>=2.32.3",
     "aiohttp>=3.10.11",
-    "pillow (>=11.0.0,<12.0.0)",
+    "pillow>=11.0.0",
     "openmetadata-ingestion==1.5.12",
-    "zstandard (>=0.23.0,<0.24.0)"
+    "zstandard>=0.23.0"
+]
+
+[dependency-groups]
+lint = [
+    "black>=24.8.0",
+    "flake8>=7.1.1",
+    "isort>=5.13.2",
+]
+test = [
+    "pytest>=8.3.2",
+    "pytest-cov>=5.0.0",
+]
+docs = [
+    "sphinx>=8.0.2",
+    "myst-parser>=4.0.0",
+    "sphinx_rtd_theme>=3.0.1",
+]
+dev = [
+    {include-group = "lint"},
+    {include-group = "test"},
+    {include-group = "docs"}
 ]
 
 
@@ -45,34 +66,34 @@ packages = [{include = "tatm", from = "src"}]
 optional = true
 
 [tool.poetry.group.dev.dependencies]
-black = "^24.8.0"
-flake8 = "^7.1.1"
-pytest = "^8.3.2"
-pytest-cov = "^5.0.0"
-sphinx = "^8.0.2"
-myst-parser = "^4.0.0"
-isort = "^5.13.2"
-sphinx_rtd_theme = "^3.0.1"
+black = ">=24.8.0"
+flake8 = ">=7.1.1"
+pytest = ">=8.3.2"
+pytest-cov = ">=5.0.0"
+sphinx = ">=8.0.2"
+myst-parser = ">=4.0.0"
+isort = ">=5.13.2"
+sphinx_rtd_theme = ">=3.0.1"
 
 
 [tool.poetry.group.test]
 optional = true
 
 [tool.poetry.group.test.dependencies]
-black = "^24.8.0"
-flake8 = "^7.1.1"
-pytest = "^8.3.2"
-pytest-cov = "^5.0.0"
-isort = "^5.13.2"
+black = ">=24.8.0"
+flake8 = ">=7.1.1"
+pytest = ">=8.3.2"
+pytest-cov = ">=5.0.0"
+isort = ">=5.13.2"
 
 
 [tool.poetry.group.docs]
 optional = true
 
 [tool.poetry.group.docs.dependencies]
-sphinx = "^8.0.2"
-myst-parser = "^4.0.0"
-sphinx_rtd_theme = "^3.0.1"
+sphinx = ">=8.0.2"
+myst-parser = ">=4.0.0"
+sphinx_rtd_theme = ">=3.0.1"
 
 
 [tool.isort]