wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add precision param to format_big_number

This commit is contained in:
Radek Osmulski 2024-05-31 14:35:42 +10:00
parent 57fb5fe8a6
commit af6be5c308
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/utils/utils.py
View File

@ -120,13 +120,13 @@ def init_logging():
logging.getLogger().addHandler(console_handler) logging.getLogger().addHandler(console_handler)
def format_big_number(num): def format_big_number(num, precision=0):
suffixes = ["", "K", "M", "B", "T", "Q"] suffixes = ["", "K", "M", "B", "T", "Q"]
divisor = 1000.0 divisor = 1000.0
for suffix in suffixes: for suffix in suffixes:
if abs(num) < divisor: if abs(num) < divisor:
return f"{num:.0f}{suffix}" return f"{num:.{precision}f}{suffix}"
num /= divisor num /= divisor
return num return num