pkgs/development/python-modules/face-recognition/default.nix

   1 { buildPythonPackage
   2 , fetchPypi
   3 , lib
   4
   5   # propagates
   6 , click
   7 , dlib
   8 , face-recognition-models
   9 , numpy
  10 , pillow
  11
  12   # tests
  13 , pytestCheckHook
  14 , config
  15 , cudaSupport ? config.cudaSupport
  16 }:
  17
  18 buildPythonPackage rec {
  19   pname = "face-recognition";
  20   version = "1.3.0";
  21   format = "setuptools";
  22
  23   src = fetchPypi {
  24     pname = "face_recognition";
  25     inherit version;
  26     hash = "sha256-Xl790WhqpWavDTzBMTsTHksZdleo/9A2aebT+tknBew=";
  27   };
  28
  29   propagatedBuildInputs = [
  30     click
  31     dlib
  32     face-recognition-models
  33     numpy
  34     pillow
  35   ];
  36
  37   nativeCheckInputs = [
  38     pytestCheckHook
  39   ];
  40
  41   # Disables tests when running with cuda due to https://github.com/NixOS/nixpkgs/issues/225912
  42   doCheck = !config.cudaSupport;
  43
  44   meta = with lib; {
  45     license = licenses.mit;
  46     homepage = "https://github.com/ageitgey/face_recognition";
  47     maintainers = with maintainers; [ ];
  48     description = "The world's simplest facial recognition api for Python and the command line";
  49   };
  50 }