setup.py: use use wheel location instead of wheel filename

Signed-off-by: Daniele Trifirò <[email protected]>
vllm-project · Dec 3, 2024 · b17f15b · b17f15b
1 parent 0acb1b2
commit b17f15b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -505,7 +505,8 @@ def repackage_wheel(package_data: Dict[str, List[str]],
 
     import zipfile
 
-    if os.path.exists(wheel_filename := os.path.basename(wheel_location)):
+    if os.path.isfile(wheel_location):
+        wheel_filename = os.path.realpath(wheel_location)
         print(f"Using existing wheel={wheel_filename}")
     else:
         try: