From 07f6955b94b846227d67eb33cf458c6db30fac02 Mon Sep 17 00:00:00 2001
From: Jeremy Schlatter <jeremy@jeremyschlatter.com>
Date: Wed, 26 Jun 2024 15:10:21 -0700
Subject: [PATCH] python3Packages.torch: fix tests.*.gpuCheck

---
 pkgs/development/python-modules/torch/gpu-checks.nix | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkgs/development/python-modules/torch/gpu-checks.nix b/pkgs/development/python-modules/torch/gpu-checks.nix
index d01fffe45cb0..55a4b45f7152 100644
--- a/pkgs/development/python-modules/torch/gpu-checks.nix
+++ b/pkgs/development/python-modules/torch/gpu-checks.nix
@@ -13,7 +13,7 @@ let
       torch,
       cudaPackages,
     }:
-    cudaPackages.writeGpuPythonTest
+    cudaPackages.writeGpuTestPython
       {
         inherit feature;
         libraries = [ torch ];