From bc0856df558394710a78e7364e3761c29a397164 Mon Sep 17 00:00:00 2001
From: JalenCato <jalencato23pistons@gmail.com>
Date: Tue, 12 Dec 2023 22:31:16 +0000
Subject: [PATCH] update to default

---
 python/graphstorm/config/argument.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/python/graphstorm/config/argument.py b/python/graphstorm/config/argument.py
index f1583893e0..494ddf8fbf 100644
--- a/python/graphstorm/config/argument.py
+++ b/python/graphstorm/config/argument.py
@@ -25,6 +25,7 @@
 import yaml
 import torch as th
 import torch.nn.functional as F
+from dgl.distributed.constants import DEFAULT_NTYPE, DEFAULT_ETYPE
 
 from .config import BUILTIN_GNN_ENCODER
 from .config import BUILTIN_ENCODER
@@ -149,7 +150,6 @@ def __init__(self, cmd_args):
         # Override class attributes using command-line arguments
         self.override_arguments(cmd_args)
         self.local_rank = cmd_args.local_rank
-        self.is_homo = False
 
         logging.debug(str(configuration))
         cmd_args_dict = cmd_args.__dict__
@@ -1577,8 +1577,7 @@ def target_ntype(self):
         if hasattr(self, "_target_ntype"):
             return self._target_ntype
         else:
-            self.is_homo = True
-            return "_N"
+            return DEFAULT_NTYPE
 
     @property
     def eval_target_ntype(self):