lol

python312Packages.torch: fix tests (#404474)

authored by

Gaétan Lepage and committed by
GitHub
246654b4 5a837cb8

+2 -4
+1 -2
pkgs/development/python-modules/torch/tests/mk-runtime-check.nix
··· 3 3 feature, 4 4 libraries, 5 5 versionAttr, 6 - pythonPackages, 7 6 }: 8 7 9 - (cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; }) 8 + cudaPackages.writeGpuTestPython 10 9 { 11 10 inherit feature; 12 11 inherit libraries;
+1 -2
pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix
··· 4 4 lib, 5 5 libraries, 6 6 name ? if feature == null then "torch-compile-cpu" else "torch-compile-${feature}", 7 - pythonPackages, 8 7 stdenv, 9 8 }: 10 9 let 11 10 deviceStr = if feature == null then "" else '', device="cuda"''; 12 11 in 13 - (cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; }) 12 + cudaPackages.writeGpuTestPython 14 13 { 15 14 inherit name feature libraries; 16 15 makeWrapperArgs = [