From 5bc385e73e5a55e78577b938eaa003a0d96bb5f3 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 5 May 2025 22:19:53 +0200
Subject: [PATCH] python312Packages.torch: fix tests

---
 .../python-modules/torch/tests/mk-runtime-check.nix            | 3 +--
 .../python-modules/torch/tests/mk-torch-compile-check.nix      | 3 +--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix b/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix
index 61180a19aaba..3ead20d3eeb0 100644
--- a/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix
+++ b/pkgs/development/python-modules/torch/tests/mk-runtime-check.nix
@@ -3,10 +3,9 @@
   feature,
   libraries,
   versionAttr,
-  pythonPackages,
 }:
 
-(cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; })
+cudaPackages.writeGpuTestPython
   {
     inherit feature;
     inherit libraries;
diff --git a/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix b/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix
index 268ed5297da9..9741902dab2a 100644
--- a/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix
+++ b/pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix
@@ -4,13 +4,12 @@
   lib,
   libraries,
   name ? if feature == null then "torch-compile-cpu" else "torch-compile-${feature}",
-  pythonPackages,
   stdenv,
 }:
 let
   deviceStr = if feature == null then "" else '', device="cuda"'';
 in
-(cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; })
+cudaPackages.writeGpuTestPython
   {
     inherit name feature libraries;
     makeWrapperArgs = [