fix setup (#3253)

alibaba · Sep 22, 2023 · 17b5438 · 17b5438
1 parent 1cb7b05
commit 17b5438
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/setup.py b/python/setup.py
@@ -260,7 +260,7 @@ def parsed_package_data():
 
 def build_learning_engine():
     ext_modules = [graphlearn_ext()]
-    if torch:
+    if torch and os.path.exists(glt_root_path):
         sys.path.append(
             os.path.join(glt_root_path, "graphlearn_torch", "python", "utils")
         )