diff --git a/.github/workflows/pypi.yaml b/.github/workflows/pypi.yaml
index 5fde943ac..5bbfde667 100644
--- a/.github/workflows/pypi.yaml
+++ b/.github/workflows/pypi.yaml
@@ -23,8 +23,8 @@ jobs:
       - name: Set up Python 3.11
         uses: actions/setup-python@v5
         with:
-          python-version: "3.11"
-          cache: "poetry"
+          python-version: '3.11'
+          cache: 'poetry'
 
       - name: Install Dependencies
         run: poetry env use python3.11 && poetry install
diff --git a/pyproject.toml b/pyproject.toml
index 98512c544..a4a96f618 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "validmind"
-version = "2.9.5"
+version = "2.10.0"
 description = "ValidMind Library"
 readme = "README.pypi.md"
 requires-python = ">=3.9,<3.13"
diff --git a/validmind/__version__.py b/validmind/__version__.py
index 933fa29c8..1c622223b 100644
--- a/validmind/__version__.py
+++ b/validmind/__version__.py
@@ -1 +1 @@
-__version__ = "2.9.5"
+__version__ = "2.10.0"
diff --git a/validmind/utils.py b/validmind/utils.py
index 5d8306a05..681314770 100644
--- a/validmind/utils.py
+++ b/validmind/utils.py
@@ -26,6 +26,7 @@
 from IPython.display import display as ipy_display
 from matplotlib.axes._axes import _log as matplotlib_axes_logger
 from numpy import ndarray
+from packaging.version import Version
 from sklearn.exceptions import UndefinedMetricWarning
 from tabulate import tabulate
 
@@ -73,7 +74,9 @@ def parse_version(version: str) -> tuple[int, ...]:
     Returns:
         tuple[int, ...]: A tuple of major, minor, patch integers.
     """
-    return tuple(int(x) for x in version.split(".")[:3])
+    v = Version(version)
+
+    return (v.major, v.minor, v.micro)
 
 
 def is_notebook() -> bool: