From 229171416df3b597cc3cab3df5d9144d3107669e Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Fri, 12 Jun 2020 09:44:11 +0100
Subject: [PATCH] [CustomOp] Run pre-commit on QuantAvgPool2d
---
src/finn/custom_op/quantavgpool2d.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/custom_op/quantavgpool2d.py b/src/finn/custom_op/quantavgpool2d.py
index 0a848b859..2a8d3f16d 100644
--- a/src/finn/custom_op/quantavgpool2d.py
+++ b/src/finn/custom_op/quantavgpool2d.py
@@ -34,7 +34,7 @@ class QuantAvgPool2d(CustomOp):
def infer_node_datatype(self, model):
node = self.onnx_node
bw = self.get_nodeattr("obits")
- if bw in [2,4,8,16,32]:
+ if bw in [2, 4, 8, 16, 32]:
if self.get_nodeattr("signed") == 0:
dtype = DataType["UINT%d" % bw]
else:
--
GitLab