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

Co-authored-by: Sanghun Lee <sanghun@lablup.com>
lablup · Oct 24, 2024 · 53bd292 · 53bd292
1 parent 7f56816
commit 53bd292
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
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
@@ -516,7 +516,7 @@ 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:
+            except (ArithmeticError, IndexError):
                 raise ValueError("Unconvertible value", orig_expr, ending)
             try:
                 multiplier = cls.suffix_map[suffix]