diff options
Diffstat (limited to 'verif/frameworks/arg_gen.py')
| -rw-r--r-- | verif/frameworks/arg_gen.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/verif/frameworks/arg_gen.py b/verif/frameworks/arg_gen.py index a25c205..c385274 100644 --- a/verif/frameworks/arg_gen.py +++ b/verif/frameworks/arg_gen.py @@ -45,6 +45,10 @@ class ArgGen: @staticmethod def agAxes(op, shapes, rng): axes = [] + if shapes == (): + axes.append(["_axis_0", [0]]) + return axes + for i in range(-len(shapes), len(shapes), 1): if i >= 0: axes.append(["_axis_{}".format(i), [i]]) |