add config

awslabs · Dec 15, 2023 · 009ff4f · 009ff4f
1 parent f685f7a
commit 009ff4f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/graphstorm/config/argument.py b/python/graphstorm/config/argument.py
@@ -1655,8 +1655,7 @@ def target_etype(self):
         if not hasattr(self, "_target_etype"):
             logging.warning("There is not target etype provided, "
                             "will treat as homogeneous graph")
-            print("argument: ", list(DEFAULT_ETYPE))
-            return list(DEFAULT_ETYPE)
+            return [DEFAULT_ETYPE]
         assert isinstance(self._target_etype, list), \
             "target_etype must be a list in format: " \
             "[\"query,clicks,asin\", \"query,search,asin\"]."