diff options
author | Jörg Thalheim <joerg@thalheim.io> | 2020-06-30 08:59:39 +0100 |
---|---|---|
committer | Jörg Thalheim <joerg@thalheim.io> | 2020-06-30 08:59:39 +0100 |
commit | 0c5e60202961ccea1c8ad8ac1201998e37f0791c (patch) | |
tree | f35cb36d10edcacd4550c8c18d33619bfdb33496 /pkgs/development/python-modules/tensorboardx | |
parent | 9c86c74cd6d9c4d823fee86d5ce5889dd8320897 (diff) | |
download | nixpkgs-0c5e60202961ccea1c8ad8ac1201998e37f0791c.tar nixpkgs-0c5e60202961ccea1c8ad8ac1201998e37f0791c.tar.gz nixpkgs-0c5e60202961ccea1c8ad8ac1201998e37f0791c.tar.bz2 nixpkgs-0c5e60202961ccea1c8ad8ac1201998e37f0791c.tar.lz nixpkgs-0c5e60202961ccea1c8ad8ac1201998e37f0791c.tar.xz nixpkgs-0c5e60202961ccea1c8ad8ac1201998e37f0791c.tar.zst nixpkgs-0c5e60202961ccea1c8ad8ac1201998e37f0791c.zip |
tensorboardx: use pytestCheckHook
Diffstat (limited to 'pkgs/development/python-modules/tensorboardx')
-rw-r--r-- | pkgs/development/python-modules/tensorboardx/default.nix | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/pkgs/development/python-modules/tensorboardx/default.nix b/pkgs/development/python-modules/tensorboardx/default.nix index 3c9c9f5c898..d2532777d00 100644 --- a/pkgs/development/python-modules/tensorboardx/default.nix +++ b/pkgs/development/python-modules/tensorboardx/default.nix @@ -1,6 +1,6 @@ -{ boto3, buildPythonPackage, crc32c, fetchFromGitHub, lib, matplotlib, moto, - numpy, pillow, pytorch, protobuf, six, pytest, - tensorflow-tensorboard, torchvision }: +{ boto3, buildPythonPackage, crc32c, fetchFromGitHub, lib, matplotlib, moto +, numpy, pillow, pytorch, protobuf, six, pytestCheckHook +, tensorflow-tensorboard, torchvision }: buildPythonPackage rec { pname = "tensorboardx"; @@ -13,18 +13,17 @@ buildPythonPackage rec { 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 ]; - 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 - ''; + disabledTests = [ "test_TorchVis" "test_onnx_graph" ]; meta = with lib; { description = "Library for writing tensorboard-compatible logs"; - homepage = "https://github.com/lanpa/tensorboard-pytorch"; + homepage = "https://github.com/lanpa/tensorboardX"; license = licenses.mit; maintainers = with maintainers; [ lebastr akamaus ]; platforms = platforms.all; |