wehub-resource-sync
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
import shutil
|
|
import subprocess
|
|
|
|
|
|
def install_requirements(requirements_path):
|
|
"""Install requirements.
|
|
|
|
Args:
|
|
requirements_path: TODO.
|
|
"""
|
|
try:
|
|
result = pip_install_r(requirements_path)
|
|
# check status
|
|
if result.returncode == 0:
|
|
print("install model requirements successfully")
|
|
return True
|
|
else:
|
|
print("fail to install model requirements! ")
|
|
print("error", result.stderr)
|
|
return False
|
|
except Exception as e:
|
|
result = pip_install_r(requirements_path)
|
|
# check status
|
|
if result.returncode == 0:
|
|
print("install model requirements successfully")
|
|
return True
|
|
else:
|
|
print("fail to install model requirements! ")
|
|
print("error", result.stderr)
|
|
return False
|
|
|
|
|
|
def pip_install_r(requirements_path):
|
|
"""Pip install r.
|
|
|
|
Args:
|
|
requirements_path: TODO.
|
|
"""
|
|
cmd = []
|
|
if shutil.which("pip") is not None:
|
|
cmd = ["pip"]
|
|
elif shutil.which("uv") is not None:
|
|
cmd = ["uv", "pip"]
|
|
else:
|
|
raise RuntimeError("pip not found, failed to install model requirements")
|
|
cmd += ["install", "-r", requirements_path]
|
|
return subprocess.run(
|
|
cmd,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
)
|