tjh.dev
1{ boto3, buildPythonPackage, crc32c, fetchFromGitHub, lib, matplotlib, moto
2, numpy, pillow, pytorch, protobuf, six, pytestCheckHook
3, tensorflow-tensorboard, torchvision }:
4
5buildPythonPackage rec {
6 pname = "tensorboardx";
7 version = "2.0";
8
9 src = fetchFromGitHub {
10 owner = "lanpa";
11 repo = "tensorboardX";
12 rev = "v${version}";
13 sha256 = "0qqalq0fhbx0wnd8wdwhyhkkv2brvj9qbk3373vk3wjxbribf5c7";
14 };
15
16 checkInputs = [
17 pytestCheckHook boto3 crc32c matplotlib moto pillow pytorch tensorflow-tensorboard torchvision
18 ];
19
20 propagatedBuildInputs = [ numpy protobuf six ];
21
22 disabledTests = [ "test_TorchVis" "test_onnx_graph" ];
23
24 meta = with lib; {
25 broken = true;
26 description = "Library for writing tensorboard-compatible logs";
27 homepage = "https://github.com/lanpa/tensorboardX";
28 license = licenses.mit;
29 maintainers = with maintainers; [ lebastr akamaus ];
30 platforms = platforms.all;
31 };
32}