pythonPackages.tensorboardx: init at 2.0

This commit is contained in:
Dmitry Vyal 2019-01-05 00:20:43 +03:00
parent 2633167c2b
commit 9c86c74cd6
2 changed files with 34 additions and 0 deletions

View file

@ -0,0 +1,32 @@
{ boto3, buildPythonPackage, crc32c, fetchFromGitHub, lib, matplotlib, moto,
numpy, pillow, pytorch, protobuf, six, pytest,
tensorflow-tensorboard, torchvision }:
buildPythonPackage rec {
pname = "tensorboardx";
version = "2.0";
src = fetchFromGitHub {
owner = "lanpa";
repo = "tensorboardX";
rev = "v${version}";
sha256 = "0qqalq0fhbx0wnd8wdwhyhkkv2brvj9qbk3373vk3wjxbribf5c7";
};
checkInputs = [ boto3 crc32c matplotlib moto pillow pytorch pytest tensorflow-tensorboard torchvision ];
propagatedBuildInputs = [ numpy protobuf six ];
postPatch = ''
substituteInPlace tests/test_visdom.py --replace test_TorchVis _skip_test_TorchVis
substituteInPlace tests/test_onnx_graph.py --replace test_onnx_graph _skip_test_onnx_graph
'';
meta = with lib; {
description = "Library for writing tensorboard-compatible logs";
homepage = "https://github.com/lanpa/tensorboard-pytorch";
license = licenses.mit;
maintainers = with maintainers; [ lebastr akamaus ];
platforms = platforms.all;
};
}

View file

@ -7238,6 +7238,8 @@ in {
suseapi = callPackage ../development/python-modules/suseapi { };
tensorboardx = callPackage ../development/python-modules/tensorboardx { };
typed-ast = callPackage ../development/python-modules/typed-ast { };
stripe = callPackage ../development/python-modules/stripe { };