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

[PyTorch][TorchScript] Add support for join on List of strings in TorchScript

Browse Source 
Summary: Add support for join on List of strings in TorchScript.

Test Plan:
(pytorch) smummadi@smummadi-mbp pytorch % python test/test_jit_string.py
Fail to import hypothesis in common_utils, tests are not derandomized
.
----------------------------------------------------------------------
Ran 1 test in 1.090s

OK

Differential Revision: D19611800

fbshipit-source-id: cef66356abc14dfd100a806d25dd1a8bc9af0a11
This commit is contained in:
Sampath Mummadi
2020-01-29 18:21:07 -08:00
committed by Facebook Github Bot
parent cccf5e7011
commit 8ead65a946
3 changed files with 32 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/test_jit_string.py
View File

@ -1,5 +1,5 @@
from test_jit import JitTestCase
from torch.testing._internal.common_utils import run_tests
class TestScript(JitTestCase):
def test_str_ops(self):
@ -310,3 +310,17 @@ class TestScript(JitTestCase):
for i in range(len(inputs) - 1):
self.checkScript(test_str_cmp, (inputs[i], inputs[i + 1]))
def test_str_join():
return (
",".join(["a"]),
",".join(["a", "b", "c"]),
",".join(["aa", "bb", "cc"]),
",".join(["a,a", "bb", "c,c"]),
"**a**".join(["b", "c", "d", "e"]),
"".join(["a", "b", "c"]),
)
self.checkScript(test_str_join, ())
if __name__ == '__main__':
run_tests()