summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLeaveMyYard <zhukovpavel2001@gmail.com>2023-08-17 16:43:02 +0300
committerLeaveMyYard <zhukovpavel2001@gmail.com>2023-08-17 16:43:02 +0300
commitef4c0deff3733cdc1d336b054239e20ff10ffd99 (patch)
tree79d48eaaaa7d8c980f15278a296cab48ac4591ca
parent3daa36a89f33f2934400548525f0e0f9618e8195 (diff)
Fix resource format bug
-rw-r--r--robusta_krr/utils/resource_units.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/robusta_krr/utils/resource_units.py b/robusta_krr/utils/resource_units.py
index 8489b75..9fd290d 100644
--- a/robusta_krr/utils/resource_units.py
+++ b/robusta_krr/utils/resource_units.py
@@ -42,6 +42,8 @@ def format(x: Union[float, int], /, *, base: Literal[1024, 1000] = 1024) -> str:
if x < 1:
return f"{int(x*1000)}m"
+ if x < base:
+ return str(x)
units = ["", "K", "M", "G", "T", "P", "E"]
binary_units = ["", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei"]