Skip to content

GitLab

Unverified Commit d89c0e4b authored by Kebe's avatar Kebe Committed by GitHub
Browse files

Use metadata to detect version of package (#4782) 


Signed-off-by: default avatarKebe <mail@kebe7jun.com>
parent fa3c9e06
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
python/sglang/check_env.py
View file @ d89c0e4b
"""Check environment configurations and dependency versions.""" """Check environment configurations and dependency versions."""
import importlib import importlib.metadata
import os import os
import resource import resource
import subprocess import subprocess
...@@ -59,9 +59,8 @@ def get_package_versions(packages): ...@@ -59,9 +59,8 @@ def get_package_versions(packages):
for package in packages: for package in packages:
package_name = package.split("==")[0].split(">=")[0].split("<=")[0] package_name = package.split("==")[0].split(">=")[0].split("<=")[0]
try: try:
module = importlib.import_module(package_name) version = importlib.metadata.version(package_name)
if hasattr(module, "__version__"): versions[package_name] = version
versions[package_name] = module.__version__
except ModuleNotFoundError: except ModuleNotFoundError:
versions[package_name] = "Module Not Found" versions[package_name] = "Module Not Found"
return versions return versions
......
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