fix: Handle IndexError when parse BinarySize (#2962)

lablup · Oct 24, 2024 · c72719b · c72719b
1 parent f747a61
commit c72719b
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
@@ -518,7 +518,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]