format

alibaba · Oct 11, 2023 · 0621a9a · 0621a9a
1 parent 6440c8f
commit 0621a9a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -137,9 +137,8 @@ def run(self):
 class BuildGLTorchExt(torch.utils.cpp_extension.BuildExtension if torch else build_ext):
     def run(self):
         if torch is None:
-            print(
-                "Building graphlearn-torch extension requires installing pytorch first. Let WITH_GLTORCH=OFF if you don't need it."
-            )
+            print("Building graphlearn-torch extension requires pytorch")
+            print("Set WITH_GLTORCH=OFF if you don't need it.")
             return
         self.extensions = [
             ext