Merge 17cdd0d852455dedc8546ee7b5ab444877b3f135 into 553741557a57df44a46fb3ab72b6aafeaf945337

fix: validate boolean environment variables with strict parsing (#218 )
2026-10-28 14:57:09 +08:00 · 2026-10-28 09:58:02 +08:00 · 2026-10-25 19:59:27 +08:00
1 changed files with 8 additions and 1 deletions
--- a/deepinfer/environ.py
+++ b/deepinfer/environ.py
@ -7,10 +7,17 @@ from datetime import timedelta
 from typing import Any

 DEFAULT_PREFILL_MAX_SUM_SEQLEN_K = 2**30
+_FALSE = {"0", "", "false"}
+_TRUE = {"1", "true"}


 def _bool(v: str | None, default: str = "0") -> bool:
-    return (v if v is not None else default) not in ("0", "", "false", "False")
+    value = (v if v is not None else default).lower()
+    if value in _FALSE:
+        return False
+    if value in _TRUE:
+        return True
+    raise ValueError("expected one of '0', '1', 'false', 'true' (case-insensitive)")


 def _int(v: str | None, default: int = 0) -> int:
Author	SHA1	Message	Date
Zhang	4d0e108c5a	Merge 17cdd0d852455dedc8546ee7b5ab444877b3f135 into 553741557a57df44a46fb3ab72b6aafeaf945337	2026-10-28 09:58:02 +08:00
Inference_	553741557a	fix: validate boolean environment variables with strict parsing (#218 )	2026-10-25 19:59:27 +08:00