python3Packages.accelerate: fix build, add torchvision to test dependencies (#507217)

2026-04-06 10:15:07 +00:00
parent a976f83df6 39b4a2cfbf
commit f70f42b622
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
  ];