tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 fetchFromGitHub,
3 pytestCheckHook,
4 torch,
5 buildPythonPackage,
6 lib,
7}:
8
9buildPythonPackage rec {
10 pname = "ttach";
11 version = "0.0.3";
12 format = "setuptools";
13
14 src = fetchFromGitHub {
15 owner = "qubvel";
16 repo = "ttach";
17 tag = "v${version}";
18 hash = "sha256-R6QO+9hv0eI7dZW5iJf096+LU1q+vnmOpveurgZemPc=";
19 };
20
21 propagatedBuildInputs = [ torch ];
22
23 nativeCheckInputs = [ pytestCheckHook ];
24 pythonImportsCheck = [ "ttach" ];
25
26 meta = {
27 description = "Image Test Time Augmentation with PyTorch";
28 homepage = "https://github.com/qubvel/ttach";
29 license = with lib.licenses; [ mit ];
30 };
31}