pyrox.dev
1{ fetchFromGitHub
2, pythonOlder
3, pytestCheckHook
4, torch
5, buildPythonPackage
6, lib
7}:
8
9buildPythonPackage rec {
10 pname = "ttach";
11 version = "0.0.3";
12
13 disabled = pythonOlder "3.7";
14
15 src = fetchFromGitHub {
16 owner = "qubvel";
17 repo = pname;
18 rev = "refs/tags/v${version}";
19 hash = "sha256-R6QO+9hv0eI7dZW5iJf096+LU1q+vnmOpveurgZemPc=";
20 };
21
22 propagatedBuildInputs = [ torch ];
23
24 nativeCheckInputs = [ pytestCheckHook ];
25 pythonImportsCheck = [ "ttach" ];
26
27 meta = with lib; {
28 description = "Image Test Time Augmentation with PyTorch";
29 homepage = "https://github.com/qubvel/ttach";
30 license = with licenses; [ mit ];
31 maintainers = with maintainers; [ cfhammill ];
32 };
33}