python311Packages.albumentations: test with torch{,vision}

2024-03-17 22:13:34 +01:00
parent e8ea97573f
commit 215044592e
1 changed files with 4 additions and 0 deletions
@@ -12,6 +12,8 @@
 , pytestCheckHook
 , pythonOlder
 , pythonRelaxDepsHook
+, torch
+, torchvision
 }:

 buildPythonPackage rec {
@@ -50,6 +52,8 @@ buildPythonPackage rec {
  nativeCheckInputs = [
    deepdiff
    pytestCheckHook
+    torch
+    torchvision
  ];

  disabledTests = [