From 10bc789dff780403f7458020a36929d0080344f4 Mon Sep 17 00:00:00 2001 From: Zachary DeVito Date: Sat, 13 Apr 2019 10:01:34 -0700 Subject: [PATCH] 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 --- test/test_jit.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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() -- 2.7.4