mirror of
https://github.com/SebastianWendel/nixpkgs.git
synced 2024-09-21 21:09:00 +02:00
Merge pull request #151737 from Mic92/tensorboardx
python3.pkgs.tensorboardx: fix tests
This commit is contained in:
commit
7635d29dd8
|
@ -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; {
|
||||
|
|
Loading…
Reference in a new issue