diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-02-10 17:27:34 +0000 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-02-10 17:27:34 +0000 |
commit | ae0c1c646da49096c9a33a4839b138fbde2b36b8 (patch) | |
tree | 4541f98e75ca94ed58cb55dd423823d2432d574f /verif/generator | |
parent | 830add41f1f085f63dddf8a99e2512585d06cec4 (diff) | |
download | reference_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')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 4 | ||||
-rw-r--r-- | verif/generator/tosa_verif_build_tests.py | 2 |
2 files changed, 4 insertions, 2 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 diff --git a/verif/generator/tosa_verif_build_tests.py b/verif/generator/tosa_verif_build_tests.py index d114519..69322cc 100644 --- a/verif/generator/tosa_verif_build_tests.py +++ b/verif/generator/tosa_verif_build_tests.py @@ -172,7 +172,7 @@ def parseArgs(): parser.add_argument( "--allow-pooling-and-conv-oversizes", dest="oversize", - action='store_true', + action="store_true", help="allow oversize padding, stride and kernel tests", ) |