Skip to content

Commit d9ac9e3

Browse files
authored
[Misc] fix collect_env version parse (#15267)
Signed-off-by: wangxiyuan <[email protected]>
1 parent d41faaf commit d9ac9e3

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

vllm/collect_env.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -282,11 +282,18 @@ def get_vllm_version():
282282

283283
if __version__ == "dev":
284284
return "N/A (dev)"
285-
286-
if len(__version_tuple__) == 4: # dev build
287-
git_sha = __version_tuple__[-1][1:] # type: ignore
288-
return f"{__version__} (git sha: {git_sha}"
289-
285+
version_str = __version_tuple__[-1]
286+
if isinstance(version_str, str) and version_str.startswith('g'):
287+
# it's a dev build
288+
if '.' in version_str:
289+
# it's a dev build containing local changes
290+
git_sha = version_str.split('.')[0][1:]
291+
date = version_str.split('.')[-1][1:]
292+
return f"{__version__} (git sha: {git_sha}, date: {date})"
293+
else:
294+
# it's a dev build without local changes
295+
git_sha = version_str[1:] # type: ignore
296+
return f"{__version__} (git sha: {git_sha})"
290297
return __version__
291298

292299

0 commit comments

Comments
 (0)