swendel/nixpkgs
1
0
Fork 0
mirror of https://github.com/SebastianWendel/nixpkgs.git synced 2024-09-21 21:09:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #151737 from Mic92/tensorboardx

Browse source 
python3.pkgs.tensorboardx: fix tests
This commit is contained in:
Jörg Thalheim 2022-01-09 15:44:56 +00:00 committed by GitHub
parent e4eed15d8b 19be089106
commit 7635d29dd8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
pkgs/development/python-modules/tensorboardx/default.nix
View file

@ -1,13 +1,14 @@
{ boto3
, buildPythonPackage
, crc32c
, which
, fetchFromGitHub
, lib
, matplotlib
, moto
, numpy
, pillow
, protobuf
, protobuf3_8
, pytestCheckHook
, pytorch
, six
@ -24,7 +25,7 @@ buildPythonPackage rec {
owner = "lanpa";
repo = "tensorboardX";
rev = "v${version}";
sha256 = "1kcw062bcqvqva5kag9j7q72wk3vdqgf5cnn0lxmsvhlmq5sjdfn";
sha256 = "sha256-1jWpC64Ubl07Bday4h5ue0wuDj4yPTWL2nhjtoQBnM0=";
};
# apparently torch API changed a bit at 1.6
@ -34,10 +35,16 @@ buildPythonPackage rec {
"torch.onnx.select_model_mode_for_export(model, torch.onnx.TrainingMode.EVAL)"
'';
# Wanted protobuf version is mentioned here:
# https://github.com/lanpa/tensorboardX/blob/0d08112618a2bbda4c028a15a137fed3afe77401/compile.sh#L6
nativeBuildInputs = [ which protobuf3_8 ];
# required to make tests deterministic
PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION = "python";
propagatedBuildInputs = [
crc32c
numpy
protobuf
six
soundfile
];
@ -63,9 +70,6 @@ buildPythonPackage rec {
disabledTestPaths = [
# we are not interested in linting errors
"tests/test_lint.py"
# breaks with `RuntimeError: cannot schedule new futures after interpreter shutdown`
# Upstream tracking bug: https://github.com/lanpa/tensorboardX/issues/652
"tests/test_pr_curve.py"
];
meta = with lib; {