python3Packages.torchvision: use finalAttrs

This commit is contained in:
Gaetan Lepage
2026-02-24 07:50:50 +00:00
parent b80ab338fc
commit e5e4404a1b
@@ -1,9 +1,7 @@
{
lib,
stdenv,
torch,
buildPythonPackage,
darwinMinVersionHook,
fetchFromGitHub,
# nativeBuildInputs
@@ -27,17 +25,16 @@
let
inherit (torch) cudaCapabilities cudaPackages cudaSupport;
in
buildPythonPackage.override { stdenv = torch.stdenv; } (finalAttrs: {
pname = "torchvision";
version = "0.25.0";
in
buildPythonPackage.override { stdenv = torch.stdenv; } {
format = "setuptools";
inherit pname version;
pyproject = true;
src = fetchFromGitHub {
owner = "pytorch";
repo = "vision";
tag = "v${version}";
tag = "v${finalAttrs.version}";
hash = "sha256-oktJHcT6T4f58pUO+HSBpbyS1ISH3zDlTsXQh6PcMy4=";
};
@@ -87,9 +84,9 @@ buildPythonPackage.override { stdenv = torch.stdenv; } {
meta = {
description = "PyTorch vision library";
homepage = "https://pytorch.org/";
changelog = "https://github.com/pytorch/vision/releases/tag/v${version}";
changelog = "https://github.com/pytorch/vision/releases/tag/${finalAttrs.src.tag}";
license = lib.licenses.bsd3;
platforms = with lib.platforms; linux ++ lib.optionals (!cudaSupport) darwin;
maintainers = with lib.maintainers; [ GaetanLepage ];
};
}
})