OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

fix lint (#33861)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/33861

Test Plan: Imported from OSS

Differential Revision: D20136865

Pulled By: suo

fbshipit-source-id: 4bf7ac324a0abce9b45121ac5ab438448a6f3149
This commit is contained in:
Michael Suo 2020-02-27 00:31:06 -08:00 committed by Facebook Github Bot
parent 9733711394
commit f5952cf7cb
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/jit/test_export_modes.py
View File

@ -13,7 +13,7 @@ from torch.autograd import Variable
pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(pytorch_test_dir)
from torch.testing._internal.jit_utils import JitTestCase
from torch.testing._internal.common_utils import skipIfNoLapack, suppress_warnings
from torch.testing._internal.common_utils import skipIfNoLapack
if __name__ == '__main__':
raise RuntimeError("This test file is not meant to be run directly, use:\n\n"

8
torch/nn/utils/rnn.py
View File

@ -149,8 +149,12 @@ class PackedSequence(PackedSequence_):
# TorchScript doesn't support constructors on named tuples, so we use this helper
# method to construct PackedSequence
def _packed_sequence_init_args(data, batch_sizes=None, sorted_indices=None, unsorted_indices=None):
# type: (Tensor, Optional[Tensor], Optional[Tensor], Optional[Tensor]) -> Tuple[Tensor, Tensor, Optional[Tensor], Optional[Tensor]]
def _packed_sequence_init_args(data, # type: Tensor
batch_sizes=None, # type: Optional[Tensor]
sorted_indices=None, # type: Optional[Tensor]
unsorted_indices=None # type: Optional[Tensor]
):
# type: (...) -> Tuple[Tensor, Tensor, Optional[Tensor], Optional[Tensor]]
# NB: if unsorted_indices is provided, it should be the inverse permutation
# to sorted_indices. Don't assert it here because the PackedSequence ctor
# should only be used internally.