fix DP_ENABLE_TENSORFLOW support (#3229)

Avoid installing tensorflow as build requires when `DP_ENABLE_TENSORFLOW` is `0`. Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 5, 2024 · f5bb131 · f5bb131
1 parent 7f5d67c
commit f5bb131
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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