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

Merge pull request #194083 from Mic92/tensorboardx

Browse source 
python3.pkgs.tensorboardx: fix build
This commit is contained in:
Benjamin Hipple 2022-10-02 12:20:24 -04:00 committed by GitHub
parent ec5f7d48ab 16b9b0e181
commit e5f9412eca
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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: