pyrox.dev
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}:
18let
19 pname = "fvcore";
20 version = "0.1.5.post20221221";
21 optional-dependencies = {
22 all = [ shapely ];
23 };
24in
25buildPythonPackage {
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}