diff options
author | Dmitry Vyal <dmitryvyal@gmail.com> | 2020-09-05 23:46:04 +0300 |
---|---|---|
committer | Jonathan Ringer <jonringer@users.noreply.github.com> | 2020-11-10 22:08:56 -0800 |
commit | 9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06 (patch) | |
tree | 5dc8b075c459cd663386fb93f80c476b829e3cc0 /pkgs/development/python-modules/tensorboardx | |
parent | af9b245c6e9edd3acb34f1d8447b8224b638ce89 (diff) | |
download | nixpkgs-9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06.tar nixpkgs-9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06.tar.gz nixpkgs-9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06.tar.bz2 nixpkgs-9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06.tar.lz nixpkgs-9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06.tar.xz nixpkgs-9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06.tar.zst nixpkgs-9b9cf9469ffad6c98efa980c4e12ea6d3fee4e06.zip |
python3Packages.tensorboardx: fix build after upgrade to PyTorch 1.6
Diffstat (limited to 'pkgs/development/python-modules/tensorboardx')
-rw-r--r-- | pkgs/development/python-modules/tensorboardx/default.nix | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tensorboardx/default.nix b/pkgs/development/python-modules/tensorboardx/default.nix index d2532777d00..a95f8e02928 100644 --- a/pkgs/development/python-modules/tensorboardx/default.nix +++ b/pkgs/development/python-modules/tensorboardx/default.nix @@ -19,6 +19,12 @@ buildPythonPackage rec { propagatedBuildInputs = [ numpy protobuf six ]; + # apparently torch API changed a bit at 1.6 + postPatch = '' + substituteInPlace tensorboardX/pytorch_graph.py --replace "torch.onnx.set_training(model, False)" "torch.onnx.select_model_mode_for_export(model, torch.onnx.TrainingMode.EVAL)" + ''; + + disabledTests = [ "test_TorchVis" "test_onnx_graph" ]; meta = with lib; { |