remove the error message when PT is not preinstalled

njzjz · njzjz · commit 60b690392b8b · 2024-06-20T18:44:53.000-04:00
Signed-off-by: Jinzhe Zeng &lt;jinzhe.zeng@rutgers.edu&gt;
diff --git a/backend/read_env.py b/backend/read_env.py
@@ -104,8 +104,6 @@ def get_argument_from_env() -> Tuple[str, list, list, dict, str]:
 
     if os.environ.get("DP_ENABLE_PYTORCH", "1") == "1":
         pt_install_dir, _ = find_pytorch()
-        if pt_install_dir is None:
-            raise RuntimeError("Cannot find installed PyTorch.")
         cmake_args.extend(
             [
                 "-DENABLE_PYTORCH=ON",

Original file line number	Diff line number	Diff line change
`@@ -104,8 +104,6 @@ def get_argument_from_env() -> Tuple[str, list, list, dict, str]:`
`104`	`104`
`105`	`105`	`if os.environ.get("DP_ENABLE_PYTORCH", "1") == "1":`
`106`	`106`	`pt_install_dir, _ = find_pytorch()`
`107`		`- if pt_install_dir is None:`
`108`		`- raise RuntimeError("Cannot find installed PyTorch.")`
`109`	`107`	`cmake_args.extend(`
`110`	`108`	`[`
`111`	`109`	`"-DENABLE_PYTORCH=ON",`