From b2215e83702599df17b464ae4aa87ce7bd4745a4 Mon Sep 17 00:00:00 2001
From: youkaichao <youkaichao@gmail.com>
Date: Fri, 27 Sep 2024 00:26:38 -0700
Subject: [PATCH] [misc] fix collect env (#8894)

---
 collect_env.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/collect_env.py b/collect_env.py
index c5cd8c315e749..ae7f97f355253 100644
--- a/collect_env.py
+++ b/collect_env.py
@@ -267,13 +267,23 @@ def get_neuron_sdk_version(run_lambda):
 
 
 def get_vllm_version():
+    version = ""
     try:
         import vllm
-        return vllm.__version__ + "@" + vllm.__commit__
+        version = vllm.__version__
     except Exception:
-        # old version of vllm does not have __commit__
-        return 'N/A'
-
+        pass
+    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():
     # This could be a static method if the flags are constant, or dynamic if you need to check environment variables, etc.