diff --git a/pkgs/development/python-modules/albumentations/default.nix b/pkgs/development/python-modules/albumentations/default.nix
index a7bc2d525140..4c641ca0aab0 100644
--- a/pkgs/development/python-modules/albumentations/default.nix
+++ b/pkgs/development/python-modules/albumentations/default.nix
@@ -15,8 +15,12 @@
   pydantic,
   pyyaml,
   scikit-image,
-  scikit-learn,
 
+  # optional dependencies
+  huggingface-hub,
+  pillow,
+
+  # tests
   deepdiff,
   pytestCheckHook,
   pytest-mock,
@@ -26,7 +30,7 @@
 
 buildPythonPackage rec {
   pname = "albumentations";
-  version = "1.4.11";
+  version = "1.4.16";
   pyproject = true;
 
   disabled = pythonOlder "3.8";
@@ -35,7 +39,7 @@ buildPythonPackage rec {
     owner = "albumentations-team";
     repo = "albumentations";
     rev = "refs/tags/${version}";
-    hash = "sha256-1070V9+EZ4qrhxmbMyvTbu89pLoonrn0Peb8nwp2lwA=";
+    hash = "sha256-cfft5RovAJaCNTD9P9KsAJFmdwHwE0gYQ3yd6a3AAMU=";
   };
 
   pythonRemoveDeps = [ "opencv-python" ];
@@ -50,9 +54,13 @@ buildPythonPackage rec {
     pydantic
     pyyaml
     scikit-image
-    scikit-learn
   ];
 
+  optional-dependencies = {
+    hub = [ huggingface-hub ];
+    text = [ pillow ];
+  };
+
   nativeCheckInputs = [
     deepdiff
     pytestCheckHook
@@ -62,6 +70,7 @@ buildPythonPackage rec {
   ];
 
   disabledTests = [
+    "test_pca_inverse_transform"
     # this test hangs up
     "test_transforms"
   ];