Source code for lmflow.utils.versioning
import importlib
import sys
import logging
from typing import Tuple, List, Union
from importlib.metadata import version, PackageNotFoundError
import pkg_resources
[docs]
logger = logging.getLogger(__name__)
[docs]
def get_python_version():
return sys.version_info
[docs]
def _is_package_available(package_name: str, skippable: bool = False):
assert isinstance(package_name, str), f"Invalid type of package_name: {type(package_name)}"
try:
importlib.import_module(package_name)
return True
except Exception as e:
if e.__class__ == ModuleNotFoundError:
return False
else:
if skippable:
logger.warning(f'An error occurred when importing {package_name}:\n{e}\n{package_name} is disabled.')
return False
else:
raise e
[docs]
def _is_packages_available(packages: Union[List[str], List[Tuple[str, bool]]]):
if isinstance(packages[0], str):
return all([_is_package_available(package) for package in packages])
elif isinstance(packages[0], tuple):
return all([_is_package_available(package, skippable) for package, skippable in packages])
else:
raise ValueError(f"Invalid type of packages: {type(packages[0])}")
[docs]
def is_package_version_at_least(package_name, min_version):
try:
package_version = pkg_resources.get_distribution(package_name).version
if (pkg_resources.parse_version(package_version)
< pkg_resources.parse_version(min_version)):
return False
except pkg_resources.DistributionNotFound:
return False
return True
[docs]
def is_gradio_available():
return _is_package_available("gradio")
[docs]
def is_ray_available():
return _is_package_available("ray")
[docs]
def is_vllm_available():
return _is_package_available("vllm")
[docs]
def is_flash_attn_available():
return _is_package_available("flash_attn", skippable=True)
[docs]
def is_flask_available():
return _is_packages_available(["flask", "flask_cors"])
[docs]
def is_trl_available():
return _is_package_available("trl")
[docs]
def is_multimodal_available():
return _is_packages_available(["PIL"])