immich-machine-learning: fix albumentations override

2024-09-28 06:43:20 -07:00
parent 2a2dc509fe
commit 51a4c5aacc
1 changed files with 4 additions and 1 deletions
@@ -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
+        ];
      });
    };
  };