diff options
Diffstat (limited to 'pkgs/development/python-modules/timm/default.nix')
-rw-r--r-- | pkgs/development/python-modules/timm/default.nix | 55 |
1 files changed, 29 insertions, 26 deletions
diff --git a/pkgs/development/python-modules/timm/default.nix b/pkgs/development/python-modules/timm/default.nix index 18ac893e9261d..afee885b54464 100644 --- a/pkgs/development/python-modules/timm/default.nix +++ b/pkgs/development/python-modules/timm/default.nix @@ -1,37 +1,42 @@ -{ lib -, buildPythonPackage -, pythonOlder -, fetchFromGitHub -, pdm-backend -, huggingface-hub -, pyyaml -, safetensors -, torch -, torchvision -, expecttest -, pytestCheckHook -, pytest-timeout +{ + lib, + buildPythonPackage, + pythonOlder, + fetchFromGitHub, + + # build-system + pdm-backend, + + # dependencies + huggingface-hub, + pyyaml, + safetensors, + torch, + torchvision, + + # checks + expecttest, + pytestCheckHook, + pytest-timeout, }: buildPythonPackage rec { pname = "timm"; - version = "0.9.16"; + version = "1.0.9"; pyproject = true; - disabled = pythonOlder "3.7"; + disabled = pythonOlder "3.8"; src = fetchFromGitHub { owner = "huggingface"; repo = "pytorch-image-models"; rev = "refs/tags/v${version}"; - hash = "sha256-IWEDKuI2565Z07q1MxTpzKS+CROPR6SyaD5fKcQ5eXk="; + hash = "sha256-iWZXile3hCUMx2q3VHJasX7rlJmT0OKBm9rkCXuWISw="; }; - nativeBuildInputs = [ - pdm-backend - ]; + build-system = [ pdm-backend ]; - propagatedBuildInputs = [ + dependencies = [ huggingface-hub pyyaml safetensors @@ -45,9 +50,7 @@ buildPythonPackage rec { pytest-timeout ]; - pytestFlagsArray = [ - "tests" - ]; + pytestFlagsArray = [ "tests" ]; disabledTestPaths = [ # Takes too long and also tries to download models @@ -64,11 +67,11 @@ buildPythonPackage rec { "timm.data" ]; - meta = with lib; { + meta = { description = "PyTorch image models, scripts, and pretrained weights"; homepage = "https://huggingface.co/docs/timm/index"; changelog = "https://github.com/huggingface/pytorch-image-models/blob/v${version}/README.md#whats-new"; - license = licenses.asl20; - maintainers = with maintainers; [ bcdarwin ]; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ bcdarwin ]; }; } |