Merge pull request #205245 from bcdarwin/python3-ttach

{ fetchFromGitHub

, pythonOlder

, pytestCheckHook

, torch

, buildPythonPackage

, lib

}:

buildPythonPackage rec {

  pname = "ttach";

  version = "0.0.3";

  disabled = pythonOlder "3.7";

  src = fetchFromGitHub {

    owner = "qubvel";

    repo = pname;

    rev = "refs/tags/v${version}";

    hash = "sha256-R6QO+9hv0eI7dZW5iJf096+LU1q+vnmOpveurgZemPc=";

  };

  propagatedBuildInputs = [ torch ];

  checkInputs = [ pytestCheckHook ];

  pythonImportsCheck = [ "ttach" ];

  meta = with lib; {

    description = "Image Test Time Augmentation with PyTorch";

    homepage = "https://github.com/qubvel/ttach";

    license = with licenses; [ mit ];

    maintainers = with maintainers; [ cfhammill ];

  };

}

  trytond = callPackage ../development/python-modules/trytond { };

  ttach = callPackage ../development/python-modules/ttach { };

  ttls = callPackage ../development/python-modules/ttls { };

  ttp = callPackage ../development/python-modules/ttp { };