From 2f99c33e02c597386e96a838de6bea926d767e12 Mon Sep 17 00:00:00 2001 From: Bo-Wun Cheng Date: Thu, 18 Apr 2024 15:41:19 -0700 Subject: [PATCH] fix code style again --- scripts/formatting/datastructure_sparse_ml.py | 2 +- scripts/formatting/generate_sparse_ml_formats.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/scripts/formatting/datastructure_sparse_ml.py b/scripts/formatting/datastructure_sparse_ml.py index a193a713..d3c37bad 100644 --- a/scripts/formatting/datastructure_sparse_ml.py +++ b/scripts/formatting/datastructure_sparse_ml.py @@ -58,7 +58,7 @@ def write_datastructure_bench(args, tensor, out_path, tiles=None): 'conversion on one SS tensor') parser.add_argument('-f', '--format', metavar='ssformat', type=str, action='store', help='The format that the tensor ' 'should be converted to') -parser.add_argument('-l', '--layername', type=str, default=None, help='layer name of the ml model layer this tensor belongs to') +parser.add_argument('-l', '--layername', type=str, default=None, help='layer name of the layer this tensor belongs to') parser.add_argument('-m', '--modelname', type=str, default=None, help='model namd of the ml model this tensor belongs to') parser.add_argument('-d', '--datasetname', type=str, default=None, help='name of the dataset the modle is trained on') parser.add_argument('-comb', '--combined', action='store_true', default=False, help='Whether the formatted datastructures ' diff --git a/scripts/formatting/generate_sparse_ml_formats.py b/scripts/formatting/generate_sparse_ml_formats.py index 0a9fcc51..01d43efe 100644 --- a/scripts/formatting/generate_sparse_ml_formats.py +++ b/scripts/formatting/generate_sparse_ml_formats.py @@ -66,5 +66,4 @@ def format_lookup(tensor_name, operation): "-b", kernel, "--modelname", bench, "--datasetname", dataset, - "--layername", layer], - env=formatting_env) + "--layername", layer], env=formatting_env)