depot/third_party/nixpkgs/pkgs/development/python-modules/flyingsquid/default.nix

43 lines
819 B
Nix
Raw Normal View History

{
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 ];
};
}