wehub-resource-sync
30 lines
974 B
Python
30 lines
974 B
Python
# Copyright 2018 Databricks, Inc.
|
|
import importlib.metadata
|
|
import re
|
|
|
|
VERSION = "3.14.1.dev0"
|
|
|
|
|
|
def is_release_version():
|
|
return bool(re.match(r"^\d+\.\d+\.\d+$", VERSION))
|
|
|
|
|
|
def _is_package_installed(package_name: str) -> bool:
|
|
try:
|
|
importlib.metadata.version(package_name)
|
|
return True
|
|
except importlib.metadata.PackageNotFoundError:
|
|
return False
|
|
|
|
|
|
# A flag to indicate whether the environment only has the tracing SDK
|
|
# installed, or includes the full MLflow or mlflow-skinny package.
|
|
# This is used to determine whether to import modules that require
|
|
# dependencies that are not included in the tracing SDK.
|
|
IS_TRACING_SDK_ONLY = not any(_is_package_installed(pkg) for pkg in ["mlflow", "mlflow-skinny"])
|
|
|
|
# A flag to indicate whether the environment only has the mlflow-skinny package
|
|
IS_MLFLOW_SKINNY = _is_package_installed("mlflow-skinny") and not _is_package_installed("mlflow")
|
|
|
|
IS_FULL_MLFLOW = _is_package_installed("mlflow")
|