33afbf39f6
checkInputs used to be added to nativeBuildInputs. Now we have nativeCheckInputs to do that instead. Doing this treewide change allows to keep hashes identical to before the introduction of nativeCheckInputs.
79 lines
1.5 KiB
Nix
79 lines
1.5 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, fetchFromGitHub
|
|
, cloudpickle
|
|
, scikit-learn
|
|
, scikitimage
|
|
, packaging
|
|
, psutil
|
|
, py-deprecate
|
|
, torch
|
|
, pytestCheckHook
|
|
, torchmetrics
|
|
, pytorch-lightning
|
|
}:
|
|
|
|
let
|
|
pname = "torchmetrics";
|
|
version = "0.11.0";
|
|
in
|
|
buildPythonPackage {
|
|
inherit pname version;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "PyTorchLightning";
|
|
repo = "metrics";
|
|
rev = "refs/tags/v${version}";
|
|
hash = "sha256-vHWW9TAIocU4g7++UQ0SWtOV/Cdtdq+tOV1x0MLwY7E=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
packaging
|
|
py-deprecate
|
|
];
|
|
|
|
# Let the user bring their own instance
|
|
buildInputs = [
|
|
torch
|
|
];
|
|
|
|
nativeCheckInputs = [
|
|
pytorch-lightning
|
|
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
|
|
];
|
|
};
|
|
}
|
|
|