From: Zachary DeVito Date: Sat, 13 Apr 2019 17:01:34 +0000 (-0700) Subject: fix flake8 (#19243) X-Git-Tag: accepted/tizen/6.5/unified/20211028.231830~238 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=10bc789dff780403f7458020a36929d0080344f4;p=platform%2Fupstream%2Fpytorch.git fix flake8 (#19243) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19243 ghimport-source-id: ae80aed3a5742df21afb6e55979686220a27cce7 Differential Revision: D14928670 Pulled By: zdevito fbshipit-source-id: 20ec0d5c8d6f1c515beb55e2e63eddf3b2fc12dd --- diff --git a/test/test_jit.py b/test/test_jit.py index 9f09362..1492f66 100644 --- a/test/test_jit.py +++ b/test/test_jit.py @@ -252,6 +252,7 @@ def enable_cpu_fuser(fn): torch._C._jit_override_can_fuse_on_cpu(False) return wrapper + # note: not re-entrant, use unnested only @contextmanager def disable_autodiff_subgraph_inlining(enabled=True): @@ -4507,7 +4508,7 @@ a") return z graph = _propagate_shapes(test.graph, - (torch.zeros(2, 1), torch.zeros(1, 2), torch.zeros(1, 1, 1)), False) + (torch.zeros(2, 1), torch.zeros(1, 2), torch.zeros(1, 1, 1)), False) self.assertTrue(next(graph.outputs()).type() == TensorType.get()) out_op_graph_input()