From d02eb46bfbfbc059d91dca35ec4334cc047ffb64 Mon Sep 17 00:00:00 2001
From: Connor Baker <ConnorBaker01@gmail.com>
Date: Tue, 25 Nov 2025 14:24:11 +0000
Subject: [PATCH] python3Packages.tensorrt: clean up expression

Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
---
 .../python-modules/tensorrt/default.nix       | 35 ++++++++++++++-----
 1 file changed, 27 insertions(+), 8 deletions(-)

diff --git a/pkgs/development/python-modules/tensorrt/default.nix b/pkgs/development/python-modules/tensorrt/default.nix
index 86580550c7af..4f5a2d22f4f9 100644
--- a/pkgs/development/python-modules/tensorrt/default.nix
+++ b/pkgs/development/python-modules/tensorrt/default.nix
@@ -7,12 +7,21 @@
   stdenv,
 }:
 let
-  inherit (cudaPackages.tensorrt) src pname version;
+  inherit (cudaPackages.tensorrt)
+    meta
+    pname
+    src
+    version
+    ;
   inherit (lib.versions) major minor;
   inherit (stdenv.hostPlatform) parsed;
 in
 buildPythonPackage {
-  inherit pname version;
+  # Make sure to add the cudaNamePrefix tag since we're not using cudaPackages.buildRedist but this is a
+  # redistributable.
+  pname = "${cudaPackages.cudaNamePrefix}-${pname}";
+
+  inherit version;
 
   src =
     let
@@ -31,17 +40,27 @@ buildPythonPackage {
   ];
 
   buildInputs = [
-    cudaPackages.cudnn
     cudaPackages.tensorrt
   ];
 
   pythonImportsCheck = [ "tensorrt" ];
 
-  meta = with lib; {
+  meta = {
     description = "Python bindings for TensorRT, a high-performance deep learning interface";
-    homepage = "https://developer.nvidia.com/tensorrt";
-    license = licenses.unfree;
-    platforms = [ "x86_64-linux" ];
-    broken = !(cudaPackages ? tensorrt) || !(cudaPackages ? cudnn);
+
+    # Explicitly inherit from TensorRT's meta to avoid pulling in attributes added by stdenv.mkDerivation.
+    inherit (meta)
+      badPlatforms
+      broken
+      changelog
+      downloadPage
+      homepage
+      license
+      longDescription
+      maintainers
+      platforms
+      sourceProvenance
+      teams
+      ;
   };
 }