diff --git a/pkgs/by-name/im/immich-machine-learning/package.nix b/pkgs/by-name/im/immich-machine-learning/package.nix
index 1a7255dd91ff..94cc69f50ed1 100644
--- a/pkgs/by-name/im/immich-machine-learning/package.nix
+++ b/pkgs/by-name/im/immich-machine-learning/package.nix
@@ -16,7 +16,7 @@ let
         doCheck = false;
       });
 
-      albumentations = super.albumentations.overridePythonAttrs (_: rec {
+      albumentations = super.albumentations.overridePythonAttrs (old: rec {
         version = "1.4.3";
         src = fetchFromGitHub {
           owner = "albumentations-team";
@@ -24,6 +24,9 @@ let
           rev = version;
           hash = "sha256-JIBwjYaUP4Sc1bVM/zlj45cz9OWpb/LOBsIqk1m+sQA=";
         };
+        dependencies = old.dependencies ++ [
+          self.scikit-learn
+        ];
       });
     };
   };
diff --git a/pkgs/development/python-modules/albucore/default.nix b/pkgs/development/python-modules/albucore/default.nix
index b176b817da83..07f2e3f4ce72 100644
--- a/pkgs/development/python-modules/albucore/default.nix
+++ b/pkgs/development/python-modules/albucore/default.nix
@@ -12,7 +12,7 @@
 
 buildPythonPackage rec {
   pname = "albucore";
-  version = "0.0.15";
+  version = "0.0.17";
   pyproject = true;
 
   disabled = pythonOlder "3.8";
@@ -21,7 +21,7 @@ buildPythonPackage rec {
     owner = "albumentations-team";
     repo = "albucore";
     rev = "refs/tags/${version}";
-    hash = "sha256-FA11dVw47KucSYU1+3oxw6GUQfPZtjoNPUb96vb6wqo=";
+    hash = "sha256-9fv5jewfL3JKhZyD0YS1WDNZ7wWt+8iF2DcygCOl168=";
   };
 
   pythonRemoveDeps = [ "opencv-python" ];
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"
   ];