Merge pull request #233105 from fabaff/timm-fix
python310Packages.timm: 0.6.12 -> 0.9.2
This commit is contained in:
commit
8823c6a4e8
|
@ -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"
|
||||
|
|
Loading…
Reference in a new issue