Merge pull request #3 from PINTO0309/feat/comma

thousands separator
PINTO0309 · Sep 20, 2023 · 37cf3d1 · 37cf3d1
2 parents e6955a2 + 6754999
commit 37cf3d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ssc4onnx/__init__.py b/ssc4onnx/__init__.py
@@ -1,3 +1,3 @@
 from ssc4onnx.onnx_structure_check import structure_check, main
 
-__version__ = '1.0.5'
+__version__ = '1.0.6'
diff --git a/ssc4onnx/onnx_structure_check.py b/ssc4onnx/onnx_structure_check.py
@@ -57,7 +57,7 @@ def human_readable_size(num, suffix="B"):
         if abs(num) < 1024.0:
             return f"{num:3.1f}{unit}{suffix}"
         num /= 1024.0
-    return f"{num:.1f}Yi{suffix}"
+    return f"{num:,.1f}Yi{suffix}"
 
 
 class ModelInfo:
@@ -147,10 +147,10 @@ def structure_check(
     table.add_column('OP Type')
     table.add_column('OPs')
     sorted_list = sorted(list(set(model_info.op_nums.keys())))
-    _ = [table.add_row(key, str(model_info.op_nums[key])) for key in sorted_list]
+    _ = [table.add_row(key, f"{model_info.op_nums[key]:,}") for key in sorted_list]
     table.add_row('----------------------', '----------')
     ops_count = sum([model_info.op_nums[key] for key in sorted_list])
-    table.add_row('Total number of OPs', str(ops_count))
+    table.add_row('Total number of OPs', f"{ops_count:,}")
     table.add_row('======================', '==========')
     table.add_row('Model Size', human_readable_size(model_info.model_size))
     rich_print(table)