dependency_check.py

from distutils.version import LooseVersion
from importlib import util
import logging

logger = logging.getLogger(__name__)

try:
    import tensorflow

    if LooseVersion(tensorflow.__version__) < LooseVersion("2.0.0"):
        raise ImportError()
    pstf_spec = util.find_spec("syft_tensorflow")
    tensorflow_available = pstf_spec is not None
except ImportError:
    tensorflow_available = False


tfe_spec = util.find_spec("tf_encrypted")
tfe_available = tfe_spec is not None


torch_spec = util.find_spec("torch")
torch_available = torch_spec is not None