diff options
Diffstat (limited to 'verif/frameworks/write_test_json.py')
| -rw-r--r-- | verif/frameworks/write_test_json.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/verif/frameworks/write_test_json.py b/verif/frameworks/write_test_json.py index 4e3aa40..cd42198 100644 --- a/verif/frameworks/write_test_json.py +++ b/verif/frameworks/write_test_json.py @@ -20,6 +20,7 @@ def write_test_json( framework_exclusions=None, quantized=False, test_name=None, + num_variables=None, ): test_desc = dict() @@ -74,5 +75,8 @@ def write_test_json( if quantized: test_desc["quantized"] = 1 + if num_variables: + test_desc["num_variables"] = num_variables + with open(filename, "w") as f: json.dump(test_desc, f, indent=" ") |