pkgs/development/python-modules/norfair/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5   poetry-core,
   6   filterpy,
   7   importlib-metadata,
   8   numpy,
   9   rich,
  10   scipy,
  11   motmetrics,
  12   opencv4,
  13   pytestCheckHook,
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "norfair";
  18   version = "2.2.0";
  19   pyproject = true;
  20
  21   src = fetchFromGitHub {
  22     owner = "tryolabs";
  23     repo = "norfair";
  24     rev = "v${version}";
  25     hash = "sha256-aKB5TYSLW7FOXIy9u2hK7px6eEmIQdKPrhChKaU1uYs=";
  26   };
  27
  28   nativeBuildInputs = [
  29     poetry-core
  30   ];
  31
  32   pythonRelaxDeps = [ "rich" ];
  33
  34   propagatedBuildInputs = [
  35     filterpy
  36     importlib-metadata
  37     numpy
  38     rich
  39     scipy
  40   ];
  41
  42   optional-dependencies = {
  43     metrics = [ motmetrics ];
  44     video = [ opencv4 ];
  45   };
  46
  47   nativeCheckInputs = [ pytestCheckHook ];
  48
  49   pythonImportsCheck = [ "norfair" ];
  50
  51   meta = with lib; {
  52     description = "Lightweight Python library for adding real-time multi-object tracking to any detector";
  53     changelog = "https://github.com/tryolabs/norfair/releases/tag/v${version}";
  54     homepage = "https://github.com/tryolabs/norfair";
  55     license = licenses.bsd3;
  56     maintainers = with maintainers; [ fleaz ];
  57   };
  58 }