Code clean-up using black and flake8
- No functional change
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I5ab1198b9d092cd041fa9b85b2dee9900d299bfc
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py
index df2b057..c5f4ce1 100644
--- a/ethosu/vela/weight_compressor.py
+++ b/ethosu/vela/weight_compressor.py
@@ -19,6 +19,7 @@
from collections import namedtuple
import numpy as np
+from ethosu import mlw_codec
from .data_type import DataType
from .errors import UnsupportedFeatureError
@@ -31,7 +32,6 @@
from .tensor import TensorFormat
from .tensor import TensorPurpose
from .tensor import TensorSubPurpose
-from ethosu import mlw_codec
# Contains meta info for a weight compression. If two tensors have identical weight compression config,
@@ -236,7 +236,7 @@
if tens.consumer_list[0].type == "Conv2DBackpropInputSwitchedBias":
# Transpose Convoluion, reverse weights in H and W axes
- weights = np.flip(weights, axis=(0,1))
+ weights = np.flip(weights, axis=(0, 1))
# Slice weight stream up depth-ways into bricks and compress
full_ofm_depth = quant_buf.shape[-1]