dev/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

python3Packages.pytorch-metric-learning: move cudaSupport logic to package

Browse Source 
Use the same pattern as all the majority of packages using cuda
This commit is contained in:
zimbatm 2024-07-09 14:56:39 +02:00
parent da07f98d58
commit c6424f784c
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkgs/development/python-modules/pytorch-metric-learning/default.nix
View File

@ -4,6 +4,7 @@
buildPythonPackage,
fetchFromGitHub,
isPy27,
config,
# build-system
setuptools,
@ -19,7 +20,7 @@
tensorboard,
# tests
cudaSupport,
cudaSupport ? config.cudaSupport,
pytestCheckHook,
torchvision
}:

4
pkgs/top-level/python-packages.nix
View File

@ -12812,9 +12812,7 @@ self: super: with self; {
pytorch-lightning = callPackage ../development/python-modules/pytorch-lightning { };
pytorch-metric-learning = callPackage ../development/python-modules/pytorch-metric-learning {
inherit (pkgs.config) cudaSupport;
};
pytorch-metric-learning = callPackage ../development/python-modules/pytorch-metric-learning { };
pytorch-msssim = callPackage ../development/python-modules/pytorch-msssim { };