commit c30f495dc013a73e371dd8053a0381e4707ab309
author Tim Hall <tim.hall@arm.com> Mon Jun 15 20:47:35 2020 +0100
committer Tim Hall <tim.hall@arm.com> Thu Jun 18 17:53:52 2020 +0100
tree 902ded0cf05e6d00eb0d1a1f9c1464a1052def92
parent 0fc46dc565b6991409169cc40fe9ac74237857fa

Code clean-up using black and flake8

 - No functional change

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I5ab1198b9d092cd041fa9b85b2dee9900d299bfc

ethosu/vela/npu_performance.py

diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py
index abe7134..98626de 100644
--- a/ethosu/vela/npu_performance.py
+++ b/ethosu/vela/npu_performance.py
@@ -25,7 +25,6 @@
 
 from . import numeric_util
 from .architecture_features import Block
-from .architecture_features import Kernel
 from .nn_graph import PassPlacement
 from .nn_graph import SchedulerRewrite
 from .operation import NpuBlockType
@@ -241,7 +240,9 @@
 
         ifm_tensor, _, weight_tensor, ofm_tensor = ps.get_primary_op_ifm_ifm2_weights_ofm()
 
-        if npu_block_type in set((NpuBlockType.ConvolutionMxN, NpuBlockType.ConvolutionDepthWise, NpuBlockType.Pooling)):
+        if npu_block_type in set(
+            (NpuBlockType.ConvolutionMxN, NpuBlockType.ConvolutionDepthWise, NpuBlockType.Pooling)
+        ):
             # extent the ifm to full dimension
             ifm_tensor_brick_size = tuple(numeric_util.full_shape(4, list(ifm_tensor.brick_size), 1))
             ifm_tensor_shape = numeric_util.full_shape(4, ifm_tensor.shape, 1)