pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, pythonOlder
5, pytestCheckHook
6, bokeh
7, ipython
8, matplotlib
9, numpy
10, nbconvert
11, nbformat
12}:
13
14buildPythonPackage rec {
15 pname = "livelossplot";
16 version = "0.5.4";
17
18 disabled = pythonOlder "3.6";
19
20 # version number in source is wrong in this release
21 postPatch = ''substituteInPlace ${pname}/version.py --replace "0.5.3" "0.5.4"'';
22
23 src = fetchFromGitHub {
24 owner = "stared";
25 repo = pname;
26 rev = "v${version}";
27 sha256 = "IV6YAidoqVoKvpy+LNNHTPpobiDoGX59bHqJcBtaydk=";
28 };
29
30 propagatedBuildInputs = [ bokeh ipython matplotlib numpy ];
31
32 checkInputs = [ nbconvert nbformat pytestCheckHook ];
33
34 meta = with lib; {
35 description = "Live training loss plot in Jupyter for Keras, PyTorch, and others";
36 homepage = "https://github.com/stared/livelossplot";
37 license = licenses.mit;
38 maintainers = with maintainers; [ bcdarwin ];
39 };
40}