1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 pythonOlder,
6 pgmpy,
7 torch,
8}:
9let
10 pname = "flyingsquid";
11 version = "0.0.0a0";
12in
13buildPythonPackage {
14 inherit pname version;
15 format = "setuptools";
16
17 disabled = pythonOlder "3.8";
18
19 src = fetchFromGitHub {
20 owner = "HazyResearch";
21 repo = "flyingsquid";
22 rev = "28a713a9ac501b7597c2489468ae189943d00685";
23 hash = "sha256-DPHTSxDD4EW3nrNk2fk0pKJI/8+pQ7Awywd8nxhBruo=";
24 };
25
26 propagatedBuildInputs = [
27 pgmpy
28 torch
29 ];
30
31 pythonImportsCheck = [ "flyingsquid" ];
32
33 # no tests
34 doCheck = false;
35
36 meta = with lib; {
37 description = "More interactive weak supervision with FlyingSquid";
38 homepage = "https://github.com/HazyResearch/flyingsquid";
39 license = licenses.asl20;
40 maintainers = with maintainers; [ happysalada ];
41 };
42}