cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

python3.pkgs.tensorboardx: fix build

Browse source
This commit is contained in:
Jörg Thalheim 2022-10-02 16:52:06 +02:00
parent 8cefc55ae4
commit 16b9b0e181
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -33,6 +33,10 @@ buildPythonPackage rec {
substituteInPlace tensorboardX/pytorch_graph.py --replace \ substituteInPlace tensorboardX/pytorch_graph.py --replace \
"torch.onnx.set_training(model, False)" \ "torch.onnx.set_training(model, False)" \
"torch.onnx.select_model_mode_for_export(model, torch.onnx.TrainingMode.EVAL)" "torch.onnx.select_model_mode_for_export(model, torch.onnx.TrainingMode.EVAL)"
# Version detection seems broken here, the version reported by python is
# newer than the protobuf package itself.
sed -i -e "s/'protobuf[^']*'/'protobuf'/" setup.py
''; '';
# Wanted protobuf version is mentioned here: # Wanted protobuf version is mentioned here: