From af6be5c308afa8b924731fa5f8a12cf7a79347ef Mon Sep 17 00:00:00 2001 From: Radek Osmulski Date: Fri, 31 May 2024 14:35:42 +1000 Subject: [PATCH] Add precision param to format_big_number --- lerobot/common/utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/common/utils/utils.py b/lerobot/common/utils/utils.py index 696999ad..c429efbd 100644 --- a/lerobot/common/utils/utils.py +++ b/lerobot/common/utils/utils.py @@ -120,13 +120,13 @@ def init_logging(): logging.getLogger().addHandler(console_handler) -def format_big_number(num): +def format_big_number(num, precision=0): suffixes = ["", "K", "M", "B", "T", "Q"] divisor = 1000.0 for suffix in suffixes: if abs(num) < divisor: - return f"{num:.0f}{suffix}" + return f"{num:.{precision}f}{suffix}" num /= divisor return num