aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_test_gen.py
diff options
context:
space:
mode:
authorJeremy Johnson <jeremy.johnson@arm.com>2022-02-10 17:27:34 +0000
committerJeremy Johnson <jeremy.johnson@arm.com>2022-02-10 17:27:34 +0000
commitae0c1c646da49096c9a33a4839b138fbde2b36b8 (patch)
tree4541f98e75ca94ed58cb55dd423823d2432d574f /verif/generator/tosa_test_gen.py
parent830add41f1f085f63dddf8a99e2512585d06cec4 (diff)
downloadreference_model-ae0c1c646da49096c9a33a4839b138fbde2b36b8.tar.gz
Minor formatting fixes for python
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com> Change-Id: Iff6eb54734e7a78eb22d1a64803b937063b7f65a
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r--verif/generator/tosa_test_gen.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index 94edf6f..7e4cb1d 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -902,7 +902,9 @@ class TosaArgGen:
if max(shape) < 64:
# padding must be less than the kernel size
bigPadding = bigKernel - 1
- paddings.update({x for x in itertools.product(*([[0, bigPadding]] * 4))})
+ paddings.update(
+ {x for x in itertools.product(*([[0, bigPadding]] * 4))}
+ )
# There are too many parameter combinations, so generate them sparsely,
# very sparse for negative tests