1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 setuptools,
6 pytestCheckHook,
7 bokeh,
8 ipython,
9 matplotlib,
10 nbconvert,
11 nbformat,
12}:
13
14buildPythonPackage rec {
15 pname = "livelossplot";
16 version = "0.5.6";
17 pyproject = true;
18
19 src = fetchFromGitHub {
20 owner = "stared";
21 repo = "livelossplot";
22 tag = "v${version}";
23 hash = "sha256-qC1FBFJyf2IlDIffJ5Xs89WcN/GFA/8maODhc1u2xhA=";
24 };
25
26 build-system = [ setuptools ];
27
28 dependencies = [
29 bokeh
30 matplotlib
31 ];
32
33 pythonImportsCheck = [ "livelossplot" ];
34
35 nativeCheckInputs = [
36 ipython
37 nbconvert
38 nbformat
39 pytestCheckHook
40 ];
41
42 meta = {
43 description = "Live training loss plot in Jupyter for Keras, PyTorch, and others";
44 homepage = "https://github.com/stared/livelossplot";
45 license = lib.licenses.mit;
46 maintainers = with lib.maintainers; [ bcdarwin ];
47 };
48}