nixpkgs/pkgs/development/python-modules/livelossplot/default.nix
2021-12-25 20:08:09 -05:00

41 lines
921 B
Nix

{ lib
, buildPythonPackage
, fetchFromGitHub
, pythonOlder
, pytestCheckHook
, bokeh
, ipython
, matplotlib
, numpy
, nbconvert
, nbformat
}:
buildPythonPackage rec {
pname = "livelossplot";
version = "0.5.4";
disabled = pythonOlder "3.6";
# version number in source is wrong in this release
postPatch = ''substituteInPlace ${pname}/version.py --replace "0.5.3" "0.5.4"'';
src = fetchFromGitHub {
owner = "stared";
repo = pname;
rev = "v${version}";
sha256 = "IV6YAidoqVoKvpy+LNNHTPpobiDoGX59bHqJcBtaydk=";
};
propagatedBuildInputs = [ bokeh ipython matplotlib numpy ];
checkInputs = [ nbconvert nbformat pytestCheckHook ];
meta = with lib; {
description = "Live training loss plot in Jupyter for Keras, PyTorch, and others";
homepage = "https://github.com/stared/livelossplot";
license = licenses.mit;
maintainers = with maintainers; [ bcdarwin ];
};
}