nixpkgs/pkgs/development/python-modules/flyingsquid/default.nix
2023-07-12 07:52:19 +08:00

42 lines
811 B
Nix

{ lib
, buildPythonPackage
, fetchFromGitHub
, pythonOlder
, pgmpy
, torch
}:
let
pname = "flyingsquid";
version = "0.0.0a0";
in
buildPythonPackage {
inherit pname version;
format = "setuptools";
disabled = pythonOlder "3.8";
src = fetchFromGitHub {
owner = "HazyResearch";
repo = pname;
rev = "28a713a9ac501b7597c2489468ae189943d00685";
hash = "sha256-DPHTSxDD4EW3nrNk2fk0pKJI/8+pQ7Awywd8nxhBruo=";
};
propagatedBuildInputs = [
pgmpy
torch
];
pythonImportsCheck = [ "flyingsquid" ];
# no tests
doCheck = false;
meta = with lib; {
description = "More interactive weak supervision with FlyingSquid";
homepage = "https://github.com/HazyResearch/flyingsquid";
license = licenses.asl20;
maintainers = with maintainers; [ happysalada ];
};
}