pkgs/development/python-modules/pyannote-pipeline/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , setuptools
   5 , wheel
   6 , pyannote-core
   7 , pyannote-database
   8 , pyyaml
   9 , optuna
  10 , tqdm
  11 , docopt
  12 , filelock
  13 , scikit-learn
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "pyannote-pipeline";
  18   version = "3.0.1";
  19   pyproject = true;
  20
  21   src = fetchFromGitHub {
  22     owner = "pyannote";
  23     repo = "pyannote-pipeline";
  24     rev = version;
  25     hash = "sha256-0wSgy6kbKi9Wa5dimOz34IV5/8fSwaHDMUpaBW7tm2Y=";
  26   };
  27
  28   propagatedBuildInputs = [
  29     pyannote-core
  30     pyannote-database
  31     pyyaml
  32     optuna
  33     tqdm
  34     docopt
  35     filelock
  36     scikit-learn
  37   ];
  38
  39   nativeBuildInputs = [
  40     setuptools
  41     wheel
  42   ];
  43
  44   pythonImportsCheck = [ "pyannote.pipeline" ];
  45
  46   meta = with lib; {
  47     description = "Tunable pipelines";
  48     mainProgram = "pyannote-pipeline";
  49     homepage = "https://github.com/pyannote/pyannote-pipeline";
  50     license = licenses.mit;
  51     maintainers = with maintainers; [ ];
  52   };
  53 }