diff --git a/python/pyproject.toml b/python/pyproject.toml
index 54c342596..3f3b7a039 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -25,6 +25,7 @@ dependencies = [
   "cuda-python>=13.0",
   "decord2 ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
   "datasets",
+  "distro",
   "einops",
   "fastapi",
   "flashinfer_python[cu13]==0.6.11.post1", # keep it aligned with jit-cache version in Dockerfile