Ignore C901 in index.Tensor upper bound calculation (pytorch#1033)

Summary: Pull Request resolved: pytorch#1033 Ignore C901 Reviewed By: dbort Differential Revision: D50477317 fbshipit-source-id: 18aeb77b7824b330b390ce74945be0a9d77ec385
kirklandsign · Oct 20, 2023 · a629008 · a629008
1 parent ff20f40
commit a629008
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/exir/passes/sym_shape_eval_pass.py b/exir/passes/sym_shape_eval_pass.py
@@ -34,7 +34,7 @@ def nonzero(args, kwargs) -> List[Optional[int]]:
 
 @register_upper_bound_inference(exir_ops.edge.aten.index.Tensor)
 @register_upper_bound_inference(torch.ops.aten.index.Tensor)
-def index_Tensor(args, kwargs) -> List[Optional[int]]:
+def index_Tensor(args, kwargs) -> List[Optional[int]]:  # noqa: C901
     tensor = args[0]
     indices = args[1]