From f844f24dc59dd43f1c8dc03c228b31ec646dfbd3 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 6 Jul 2026 15:52:05 +0000
Subject: [PATCH] python3Packages.ale-py: add missing nvcc to fix CUDA build

---
 pkgs/development/python-modules/ale-py/default.nix | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/pkgs/development/python-modules/ale-py/default.nix b/pkgs/development/python-modules/ale-py/default.nix
index ec10c5decc26..e10dd2713436 100644
--- a/pkgs/development/python-modules/ale-py/default.nix
+++ b/pkgs/development/python-modules/ale-py/default.nix
@@ -1,6 +1,7 @@
 {
   lib,
   stdenv,
+  config,
   buildPythonPackage,
   fetchFromGitHub,
 
@@ -17,6 +18,9 @@
   opencv,
   zlib,
 
+  # nativeBuildInputs
+  cudaPackages,
+
   # dependencies
   numpy,
 
@@ -49,6 +53,11 @@ buildPythonPackage (finalAttrs: {
     jax
   ];
 
+  nativeBuildInputs = lib.optionals config.cudaSupport [
+    # Required by opencv's cmake
+    cudaPackages.cuda_nvcc
+  ];
+
   # disable lto on darwin, cmake cannot find llvm-ar
   postPatch = lib.optionalString stdenv.hostPlatform.isDarwin ''
     substituteInPlace src/ale/CMakeLists.txt \