parent
f6abbe3b32
commit
defebc5bb9
@ -0,0 +1,76 @@ |
||||
{ lib |
||||
, buildPythonPackage |
||||
, fetchFromGitHub |
||||
, cloudpickle |
||||
, scikit-learn |
||||
, scikitimage |
||||
, packaging |
||||
, psutil |
||||
, py-deprecate |
||||
, pytorch |
||||
, pytestCheckHook |
||||
, torchmetrics |
||||
}: |
||||
|
||||
let |
||||
pname = "torchmetrics"; |
||||
version = "0.8.1"; |
||||
in |
||||
buildPythonPackage { |
||||
inherit pname version; |
||||
|
||||
src = fetchFromGitHub { |
||||
owner = "PyTorchLightning"; |
||||
repo = "metrics"; |
||||
rev = "v${version}"; |
||||
hash = "sha256-AryEhYAeC97dO2pgHoz0Y9F//DVdX6RfCa80gI56iz4="; |
||||
}; |
||||
|
||||
propagatedBuildInputs = [ |
||||
packaging |
||||
py-deprecate |
||||
]; |
||||
|
||||
# Let the user bring their own instance |
||||
buildInputs = [ |
||||
pytorch |
||||
]; |
||||
|
||||
checkInputs = [ |
||||
scikit-learn |
||||
scikitimage |
||||
cloudpickle |
||||
psutil |
||||
pytestCheckHook |
||||
]; |
||||
|
||||
# A cyclic dependency in: integrations/test_lightning.py |
||||
doCheck = false; |
||||
passthru.tests.check = torchmetrics.overridePythonAttrs (_: { |
||||
doCheck = true; |
||||
}); |
||||
|
||||
disabledTestPaths = [ |
||||
# These require too many "leftpad-level" dependencies |
||||
"tests/text" |
||||
"tests/audio" |
||||
"tests/image" |
||||
|
||||
# A few non-deterministic things like test_check_compute_groups_is_faster |
||||
"tests/bases/test_collections.py" |
||||
]; |
||||
|
||||
pythonImportsCheck = [ |
||||
"torchmetrics" |
||||
]; |
||||
|
||||
meta = with lib; { |
||||
description = "Machine learning metrics for distributed, scalable PyTorch applications (used in pytorch-lightning)"; |
||||
homepage = "https://torchmetrics.readthedocs.io"; |
||||
license = licenses.asl20; |
||||
maintainers = with maintainers; [ |
||||
SomeoneSerge |
||||
]; |
||||
}; |
||||
} |
||||
|
Loading…
Reference in new issue