From 51a4c5aacc3bf696729e032dde5a1ef6f4d5c64c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Robert=20Sch=C3=BCtz?= <mail@dotlambda.de>
Date: Fri, 27 Sep 2024 19:32:16 -0700
Subject: [PATCH] immich-machine-learning: fix albumentations override

---
 pkgs/by-name/im/immich-machine-learning/package.nix | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pkgs/by-name/im/immich-machine-learning/package.nix b/pkgs/by-name/im/immich-machine-learning/package.nix
index 5b35ecfaea04..c24e6672e31b 100644
--- a/pkgs/by-name/im/immich-machine-learning/package.nix
+++ b/pkgs/by-name/im/immich-machine-learning/package.nix
@@ -15,7 +15,7 @@ let
         doCheck = false;
       });
 
-      albumentations = super.albumentations.overridePythonAttrs (_: rec {
+      albumentations = super.albumentations.overridePythonAttrs (old: rec {
         version = "1.4.3";
         src = fetchFromGitHub {
           owner = "albumentations-team";
@@ -23,6 +23,9 @@ let
           rev = version;
           hash = "sha256-JIBwjYaUP4Sc1bVM/zlj45cz9OWpb/LOBsIqk1m+sQA=";
         };
+        dependencies = old.dependencies ++ [
+          self.scikit-learn
+        ];
       });
     };
   };