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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , pythonOlder
   5 , pgmpy
   6 , torch
   7 }:
   8 let
   9   pname = "flyingsquid";
  10   version = "0.0.0a0";
  11 in
  12 buildPythonPackage {
  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 }