public
/
nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tensorboardx: use pytestCheckHook

This commit is contained in:
Jörg Thalheim 2020-06-30 08:59:39 +01:00
parent 9c86c74cd6
commit 0c5e602029
No known key found for this signature in database
GPG Key ID: 003F2096411B5F92
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pkgs/development/python-modules/tensorboardx/default.nix
View File

@ -1,6 +1,6 @@
{ boto3, buildPythonPackage, crc32c, fetchFromGitHub, lib, matplotlib, moto, { boto3, buildPythonPackage, crc32c, fetchFromGitHub, lib, matplotlib, moto
numpy, pillow, pytorch, protobuf, six, pytest, , numpy, pillow, pytorch, protobuf, six, pytestCheckHook
tensorflow-tensorboard, torchvision }: , tensorflow-tensorboard, torchvision }:
buildPythonPackage rec { buildPythonPackage rec {
pname = "tensorboardx"; pname = "tensorboardx";
@ -13,18 +13,17 @@ buildPythonPackage rec {
sha256 = "0qqalq0fhbx0wnd8wdwhyhkkv2brvj9qbk3373vk3wjxbribf5c7"; sha256 = "0qqalq0fhbx0wnd8wdwhyhkkv2brvj9qbk3373vk3wjxbribf5c7";
}; };
checkInputs = [ boto3 crc32c matplotlib moto pillow pytorch pytest tensorflow-tensorboard torchvision ]; checkInputs = [
pytestCheckHook boto3 crc32c matplotlib moto pillow pytorch tensorflow-tensorboard torchvision
];
propagatedBuildInputs = [ numpy protobuf six ]; propagatedBuildInputs = [ numpy protobuf six ];
postPatch = '' disabledTests = [ "test_TorchVis" "test_onnx_graph" ];
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; { meta = with lib; {
description = "Library for writing tensorboard-compatible logs"; description = "Library for writing tensorboard-compatible logs";
homepage = "https://github.com/lanpa/tensorboard-pytorch"; homepage = "https://github.com/lanpa/tensorboardX";
license = licenses.mit; license = licenses.mit;
maintainers = with maintainers; [ lebastr akamaus ]; maintainers = with maintainers; [ lebastr akamaus ];
platforms = platforms.all; platforms = platforms.all;