update to default

awslabs · Dec 12, 2023 · bc0856d · bc0856d
1 parent a02b41e
commit bc0856d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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):