fix code style again

weiya711 · Apr 18, 2024 · 2f99c33 · 2f99c33
1 parent f282425
commit 2f99c33
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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)