frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix invalidArguments for functions with tuple outputs, but no other (#468)

Browse Source 
arguments.

For example:

   >>> torch.randn(5, 5).geqrf('invalid arg')
   TypeError: geqrf received an invalid combination of arguments - got (str), but expected ()
This commit is contained in:
Sam Gross
2017-01-17 23:14:40 -05:00
committed by Soumith Chintala
parent a09f653f52
commit 473e795277
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/csrc/utils.cpp
View File

@ -301,9 +301,12 @@ std::pair<Option, std::string> _parseOption(const std::string& _option_str,
printable_option.erase(out_pos);
printable_option += "*, ";
printable_option += kwonly_part;
} else {
} else if (out_pos >= 2) {
printable_option.erase(out_pos-2);
printable_option += ")";
} else {
printable_option.erase(out_pos);
printable_option += ")";
}
has_out = true;
}