python3Packages.torch: fix tests.*.gpuCheck

This commit is contained in:
Jeremy Schlatter
2024-06-26 15:10:21 -07:00
parent 72f92f567f
commit 07f6955b94
@@ -13,7 +13,7 @@ let
torch,
cudaPackages,
}:
cudaPackages.writeGpuPythonTest
cudaPackages.writeGpuTestPython
{
inherit feature;
libraries = [ torch ];