fix: Handle IndexError when parse BinarySize (#2962) (#2964)

Co-authored-by: Sanghun Lee <sanghun@lablup.com>
lablup · Oct 24, 2024 · 602cc9c · 602cc9c
1 parent f68d98f
commit 602cc9c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/changes/2962.fix.md b/changes/2962.fix.md
@@ -0,0 +1 @@
+Handle `IndexError` when parse string to `BinarySize`
diff --git a/src/ai/backend/common/types.py b/src/ai/backend/common/types.py
@@ -529,8 +529,8 @@ def _parse_str(cls, expr: str) -> Union[BinarySize, Decimal]:
                         # has no suffix and is not an integer
                         # -> fractional bytes (e.g., 1.5 byte)
                         raise ValueError("Fractional bytes are not allowed")
-            except ArithmeticError:
-                raise ValueError("Unconvertible value", orig_expr)
+            except (ArithmeticError, IndexError):
+                raise ValueError("Unconvertible value", orig_expr, ending)
             try:
                 multiplier = cls.suffix_map[suffix]
             except KeyError: