diff --git a/backend/find_tensorflow.py b/backend/find_tensorflow.py index b43b32f954..c4d58ea0cd 100644 --- a/backend/find_tensorflow.py +++ b/backend/find_tensorflow.py @@ -47,6 +47,8 @@ def find_tensorflow() -> Tuple[Optional[str], List[str]]: list of str TensorFlow requirement if not found. Empty if found. """ + if os.environ.get("DP_ENABLE_TENSORFLOW", "1") == "0": + return None, [] requires = [] tf_spec = None