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