python3Packages.torchcodec: inherit cudaSupport from torch

2026-03-10 09:34:40 +01:00
parent 1906335e58
commit a6e30bef16
1 changed files with 2 additions and 3 deletions
@@ -1,7 +1,6 @@
 {
  lib,
  stdenv,
-  config,
  buildPythonPackage,
  fetchFromGitHub,

@@ -10,7 +9,6 @@

  # buildInputs
  ffmpeg,
-  cudaPackages,

  # build-system
  cmake,
@@ -21,7 +19,8 @@
  pytestCheckHook,
  torchvision,

-  cudaSupport ? config.cudaSupport,
+  cudaSupport ? torch.cudaSupport,
+  cudaPackages,
 }:

 buildPythonPackage rec {