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