python310Packages.timm: 0.6.12 -> 0.9.2
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:
parent
b7a9deb5bd
commit
6add9f8829
|
@ -7,33 +7,52 @@
|
||||||
, pytest-timeout
|
, pytest-timeout
|
||||||
, huggingface-hub
|
, huggingface-hub
|
||||||
, pyyaml
|
, pyyaml
|
||||||
|
, safetensors
|
||||||
, torch
|
, torch
|
||||||
, torchvision
|
, torchvision
|
||||||
}:
|
}:
|
||||||
|
|
||||||
buildPythonPackage rec {
|
buildPythonPackage rec {
|
||||||
pname = "timm";
|
pname = "timm";
|
||||||
version = "0.6.12";
|
version = "0.9.2";
|
||||||
disabled = pythonOlder "3.6";
|
format = "setuptools";
|
||||||
|
|
||||||
|
disabled = pythonOlder "3.7";
|
||||||
|
|
||||||
src = fetchFromGitHub {
|
src = fetchFromGitHub {
|
||||||
owner = "huggingface";
|
owner = "huggingface";
|
||||||
repo = "pytorch-image-models";
|
repo = "pytorch-image-models";
|
||||||
rev = "refs/tags/v${version}";
|
rev = "refs/tags/v${version}";
|
||||||
hash = "sha256-RNjCcCnNhtr5a+29Bx+k427a03MSooqvnuiDQ8cT8FA=";
|
hash = "sha256-gYrc8ds6urZvwDsTnzPjxjSTiAGzUD3RlCf0wogCrDI=";
|
||||||
};
|
};
|
||||||
|
|
||||||
propagatedBuildInputs = [
|
propagatedBuildInputs = [
|
||||||
huggingface-hub
|
huggingface-hub
|
||||||
pyyaml
|
pyyaml
|
||||||
|
safetensors
|
||||||
torch
|
torch
|
||||||
torchvision
|
torchvision
|
||||||
];
|
];
|
||||||
|
|
||||||
nativeCheckInputs = [ expecttest pytestCheckHook pytest-timeout ];
|
nativeCheckInputs = [
|
||||||
pytestFlagsArray = [ "tests" ];
|
expecttest
|
||||||
# takes too long and also tries to download models:
|
pytestCheckHook
|
||||||
disabledTestPaths = [ "tests/test_models.py" ];
|
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 = [
|
pythonImportsCheck = [
|
||||||
"timm"
|
"timm"
|
||||||
|
|
Loading…
Reference in a new issue