Unverified Commit dc4e3df5 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[misc] fix collect env (#8894)

parent 3b00b9c2
...@@ -267,13 +267,23 @@ def get_neuron_sdk_version(run_lambda): ...@@ -267,13 +267,23 @@ def get_neuron_sdk_version(run_lambda):
def get_vllm_version(): def get_vllm_version():
version = ""
try: try:
import vllm import vllm
return vllm.__version__ + "@" + vllm.__commit__ version = vllm.__version__
except Exception: except Exception:
# old version of vllm does not have __commit__ pass
return 'N/A' commit = ""
try:
import vllm
commit = vllm.__commit__
except Exception:
pass
if version != "" and commit != "":
return f"{version}@{commit}"
if version == "" and commit == "":
return "N/A"
return version or commit
def summarize_vllm_build_flags(): def summarize_vllm_build_flags():
# This could be a static method if the flags are constant, or dynamic if you need to check environment variables, etc. # This could be a static method if the flags are constant, or dynamic if you need to check environment variables, etc.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment