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

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , pythonOlder
   5 # build inputs
   6 , numpy
   7 , yacs
   8 , pyyaml
   9 , tqdm
  10 , termcolor
  11 , pillow
  12 , tabulate
  13 , iopath
  14 , shapely
  15 # check inputs
  16 , torch
  17 }:
  18 let
  19   pname = "fvcore";
  20   version = "0.1.5.post20221221";
  21   optional-dependencies = {
  22     all = [ shapely ];
  23   };
  24 in
  25 buildPythonPackage {
  26   inherit pname version;
  27   format = "setuptools";
  28
  29   disabled = pythonOlder "3.7";
  30
  31   src = fetchPypi {
  32     inherit pname version;
  33     hash = "sha256-8vsLuQVyrmUcEceOIEk+0ZsiQFUKfku7LW3oe90DeGA=";
  34   };
  35
  36   propagatedBuildInputs = [
  37     numpy
  38     yacs
  39     pyyaml
  40     tqdm
  41     termcolor
  42     pillow
  43     tabulate
  44     iopath
  45   ];
  46
  47   nativeCheckInputs = [
  48     torch
  49   ];
  50
  51   # TypeError: flop_count() missing 2 required positional arguments: 'model' and 'inputs'
  52   doCheck = false;
  53
  54   pythonImportsCheck = [
  55     "fvcore"
  56   ];
  57
  58   passthru.optional-dependencies = optional-dependencies;
  59
  60    meta = with lib; {
  61     description = "Collection of common code that's shared among different research projects in FAIR computer vision team";
  62     homepage = "https://github.com/facebookresearch/fvcore";
  63     license = licenses.asl20;
  64     maintainers = with maintainers; [ happysalada ];
  65   };
  66 }