diff --git a/poetry.lock b/poetry.lock
index fd2770dfd..dfba57e5d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,4 +1,4 @@
-# This file is automatically @generated by Poetry 1.8.2 and should not be changed by hand.
+# This file is automatically @generated by Poetry 1.6.1 and should not be changed by hand.
 
 [[package]]
 name = "albucore"
@@ -513,7 +513,7 @@ colors = ["colorama (>=0.4.6)"]
 name = "jinja2"
 version = "3.1.4"
 description = "A very fast and expressive template engine."
-optional = false
+optional = true
 python-versions = ">=3.7"
 files = [
     {file = "jinja2-3.1.4-py3-none-any.whl", hash = "sha256:bc5dd2abb727a5319567b7a813e6a2e7318c39f4f487cfe6c89c6f9c7d25197d"},
@@ -721,7 +721,7 @@ testing = ["coverage", "pytest", "pytest-cov", "pytest-regressions"]
 name = "markupsafe"
 version = "2.1.5"
 description = "Safely add untrusted strings to HTML/XML markup."
-optional = false
+optional = true
 python-versions = ">=3.7"
 files = [
     {file = "MarkupSafe-2.1.5-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:a17a92de5231666cfbe003f0e4b9b3a7ae3afb1ec2845aadc2bacc93ff85febc"},
@@ -1472,7 +1472,6 @@ files = [
     {file = "PyYAML-6.0.1-cp311-cp311-win_amd64.whl", hash = "sha256:bf07ee2fef7014951eeb99f56f39c9bb4af143d8aa3c21b1677805985307da34"},
     {file = "PyYAML-6.0.1-cp312-cp312-macosx_10_9_x86_64.whl", hash = "sha256:855fb52b0dc35af121542a76b9a84f8d1cd886ea97c84703eaa6d88e37a2ad28"},
     {file = "PyYAML-6.0.1-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:40df9b996c2b73138957fe23a16a4f0ba614f4c0efce1e9406a184b6d07fa3a9"},
-    {file = "PyYAML-6.0.1-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:a08c6f0fe150303c1c6b71ebcd7213c2858041a7e01975da3a99aed1e7a378ef"},
     {file = "PyYAML-6.0.1-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:6c22bec3fbe2524cde73d7ada88f6566758a8f7227bfbf93a408a9d86bcc12a0"},
     {file = "PyYAML-6.0.1-cp312-cp312-musllinux_1_1_x86_64.whl", hash = "sha256:8d4e9c88387b0f5c7d5f281e55304de64cf7f9c0021a3525bd3b1c542da3b0e4"},
     {file = "PyYAML-6.0.1-cp312-cp312-win32.whl", hash = "sha256:d483d2cdf104e7c9fa60c544d92981f12ad66a457afae824d146093b8c294c54"},
@@ -2156,12 +2155,12 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 
 [extras]
 dev = ["black", "debugpy", "isort", "mypy", "pytest", "pytest-rerunfailures", "responses", "ruff", "validate-pyproject"]
-medical = ["connected-components-3d", "nibabel"]
-ml = ["albumentations", "scikit-learn", "torch", "torchvision"]
+medical = ["connected-components-3d", "nibabel", "scipy"]
+ml = ["albumentations", "scikit-learn", "scipy", "torch", "torchvision"]
 ocv = ["opencv-python-headless"]
 test = ["pytest", "responses"]
 
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.8.0,<3.12"
-content-hash = "f0208e4c6d8017c91f4ff1d793f43e596d786fef0c2ef96fe5c7bc1f6bfbef67"
+content-hash = "e2acd10369632e81157fc873add2de947edd824f1f9ed2253b7d9598b3f965d5"
diff --git a/pyproject.toml b/pyproject.toml
index f311b53d7..85c8c17a4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -110,6 +110,11 @@ optional = true
 python = ">=3.8.1,<3.12"
 version = "^1.2.0"
 
+[tool.poetry.dependencies.scipy]
+optional = true
+python = ">=3.8.1,<3.12"
+version = "^1.10.1"
+
 [tool.poetry.dependencies.albumentations]
 optional = true
 python = ">=3.8"
@@ -158,7 +163,6 @@ version = "^1.0.0"
 [tool.poetry.dependencies.opencv-python-headless]
 optional = true
 version = "^4.8.0.76"
-
 [tool.poetry.dependencies.pytest-rerunfailures]
 optional = true
 version = "^12.0"