b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

python310Packages.timm: 0.6.12 -> 0.9.2

Browse source 
Diff: https://github.com/huggingface/pytorch-image-models/compare/refs/tags/v0.6.12...v0.9.2

Changelog: https://github.com/huggingface/pytorch-image-models/blob/v0.9.2/README.md#whats-new
This commit is contained in:
Fabian Affolter 2023-05-21 00:46:04 +02:00
parent b7a9deb5bd
commit 6add9f8829
1 changed files with 26 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
pkgs/development/python-modules/timm/default.nix
View file

@ -7,33 +7,52 @@
, pytest-timeout
, huggingface-hub
, pyyaml
, safetensors
, torch
, torchvision
}:
buildPythonPackage rec {
pname = "timm";
version = "0.6.12";
disabled = pythonOlder "3.6";
version = "0.9.2";
format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchFromGitHub {
owner = "huggingface";
repo = "pytorch-image-models";
rev = "refs/tags/v${version}";
hash = "sha256-RNjCcCnNhtr5a+29Bx+k427a03MSooqvnuiDQ8cT8FA=";
hash = "sha256-gYrc8ds6urZvwDsTnzPjxjSTiAGzUD3RlCf0wogCrDI=";
};
propagatedBuildInputs = [
huggingface-hub
pyyaml
safetensors
torch
torchvision
];
nativeCheckInputs = [ expecttest pytestCheckHook pytest-timeout ];
pytestFlagsArray = [ "tests" ];
# takes too long and also tries to download models:
disabledTestPaths = [ "tests/test_models.py" ];
nativeCheckInputs = [
expecttest
pytestCheckHook
pytest-timeout
];
pytestFlagsArray = [
"tests"
];
disabledTestPaths = [
# Takes too long and also tries to download models
"tests/test_models.py"
];
disabledTests = [
# AttributeError: 'Lookahead' object has no attribute '_optimizer_step_pre...
"test_lookahead"
];
pythonImportsCheck = [
"timm"