From 071d05521a967190957a28e5534c11920f0b0a24 Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-172-31-0-214.us-west-2.compute.internal>
Date: Fri, 6 Sep 2024 06:03:40 +0000
Subject: [PATCH] remove main args

---
 examples/graphbolt/pyg/hetero/node_classification.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py
index 21fdf1203899..871d74df84ed 100644
--- a/examples/graphbolt/pyg/hetero/node_classification.py
+++ b/examples/graphbolt/pyg/hetero/node_classification.py
@@ -411,7 +411,7 @@ def parse_args():
     return parser.parse_args()
 
 
-def main(args):
+def main():
     torch.set_float32_matmul_precision(args.precision)
     if not torch.cuda.is_available():
         args.mode = "cpu-cpu-cpu"
@@ -528,4 +528,4 @@ def main(args):
 
 if __name__ == "__main__":
     args = parse_args()
-    main(args)
+    main()