python3Packages.accelerate: add torchvision to test dependencies

2026-04-06 19:05:02 +10:00
parent 588f70263b
commit 39b4a2cfbf
1 changed files with 2 additions and 0 deletions
@@ -25,6 +25,7 @@
  evaluate,
  parameterized,
  pytestCheckHook,
+  torchvision,
  transformers,
  config,
  cudatoolkit,
@@ -62,6 +63,7 @@ buildPythonPackage (finalAttrs: {
    evaluate
    parameterized
    pytestCheckHook
+    torchvision
    transformers
    writableTmpDirAsHomeHook
  ];