diff options
| author | Mikhail Zolotukhin <mvz@fb.com> | 2019-01-09 16:57:22 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-09 17:01:20 -0800 |
| commit | 159c2f39189bfb95a50548c008f40ac523fc752d (patch) | |
| tree | b1e02c8f136ed49cc804653e139318943f8cccd5 /test | |
| parent | b28738ccb5b1cbf2c0ed3e6db5765aa4abc54ec3 (diff) | |
| download | pytorch-159c2f39189bfb95a50548c008f40ac523fc752d.tar.gz pytorch-159c2f39189bfb95a50548c008f40ac523fc752d.tar.bz2 pytorch-159c2f39189bfb95a50548c008f40ac523fc752d.zip | |
test_jit.py: Replace direct `exec` invocation with a wrapper. (#15882)
Summary:
Python2 doesn't allow to invoke `exec` from a nested function:
File "test/test_jit.py", line 4653
exec(code, globals(), scope)
SyntaxError: unqualified exec is not allowed in function 'test' it is a nested function
This patch wraps exec with a separate function, making it work for both python2
and python3.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15882
Differential Revision: D13614235
Pulled By: ZolotukhinM
fbshipit-source-id: 9a074308c2379f089402e0bf5a996cc649d6dbca
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_jit.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/test/test_jit.py b/test/test_jit.py index 5d959bd289..61027423da 100644 --- a/test/test_jit.py +++ b/test/test_jit.py @@ -2651,6 +2651,11 @@ class TestBatched(TestCase): w_hi, w_hf, w_ho, w_hc, b_i, b_f, b_o, b_c, w_hs, b_s, iter_num_batch, idx_batch) self.assertEqual(ys, ybs.examples()) +def execWrapper(code, glob, loc): + if PY2: + exec(code) in glob, loc + else: + exec(code, glob, loc) class TestScript(JitTestCase): @contextmanager @@ -4411,7 +4416,7 @@ a") def run_test(code): scope = {} - exec(code, globals(), scope) + execWrapper(code, globals(), scope) cu = torch.jit.CompilationUnit(code) self.assertEqual(cu.func(), scope['func']()) @@ -4483,7 +4488,7 @@ a") code = template.format(lhs=args[0], rhs=args[1], op=op) scope = {} - exec(code, globals(), scope) + execWrapper(code, globals(), scope) cu = torch.jit.CompilationUnit(code) self.assertEqual(cu.func(tensor), scope['func'](tensor)) @@ -4617,7 +4622,7 @@ a") def test(op, args): code = template.format(lhs=args[0], rhs=args[1], op=op) scope = {} - exec(code, globals(), scope) + execWrapper(code, globals(), scope) cu = torch.jit.CompilationUnit(code) self.assertEqual( cu.func(), @@ -4644,7 +4649,7 @@ a") def test(inp, typ, type_hint): code = template.format(typ=typ, type_hint=type_hint) scope = {} - exec(code, globals(), scope) + execWrapper(code, globals(), scope) cu = torch.jit.CompilationUnit(code) self.assertEqual( cu.func(inp), |