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

tensorboardx: fix tests

Browse Source
This commit is contained in:
Jörg Thalheim 2021-12-22 12:00:11 +01:00
parent f32e876f36
commit 19be089106
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,13 +1,14 @@
{ boto3 { boto3
, buildPythonPackage , buildPythonPackage
, crc32c , crc32c
, which
, fetchFromGitHub , fetchFromGitHub
, lib , lib
, matplotlib , matplotlib
, moto , moto
, numpy , numpy
, pillow , pillow
, protobuf , protobuf3_8
, pytestCheckHook , pytestCheckHook
, pytorch , pytorch
, six , six
@ -24,7 +25,7 @@ buildPythonPackage rec {
owner = "lanpa"; owner = "lanpa";
repo = "tensorboardX"; repo = "tensorboardX";
rev = "v${version}"; rev = "v${version}";
sha256 = "1kcw062bcqvqva5kag9j7q72wk3vdqgf5cnn0lxmsvhlmq5sjdfn"; sha256 = "sha256-1jWpC64Ubl07Bday4h5ue0wuDj4yPTWL2nhjtoQBnM0=";
}; };
# apparently torch API changed a bit at 1.6 # 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)" "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 = [ propagatedBuildInputs = [
crc32c crc32c
numpy numpy
protobuf
six six
soundfile soundfile
]; ];