classname

python/dgl/graphbolt/impl/ondisk_dataset.py

@@ -340,7 +340,7 @@ def test_set(self) -> Union[ItemSet, ItemSetDict]:
         return self._test_set
 
     def __repr__(self) -> str:
-        ret = "OnDiskTask({attributes})"
+        ret = "{Classname}({attributes})"
 
         attributes_str = ""
 
@@ -355,7 +355,9 @@ def __repr__(self) -> str:
             attributes_str, " " * len("OnDiskTask(")
         ).strip()
 
-        return ret.format(attributes=attributes_str)
+        return ret.format(
+            Classname=self.__class__.__name__, attributes=attributes_str
+        )
 
 
 class OnDiskDataset(Dataset):

python/dgl/graphbolt/impl/torch_based_feature_store.py

@@ -172,7 +172,7 @@ def pin_memory_(self):
 
     def __repr__(self) -> str:
         ret = (
-            "TorchBasedFeature(\n"
+            "{Classname}(\n"
             "    feature={feature},\n"
             "    metadata={metadata},\n"
             ")"
@@ -185,7 +185,11 @@ def __repr__(self) -> str:
             str(self.metadata()), " " * len("    metadata=")
         ).strip()
 
-        return ret.format(feature=feature_str, metadata=metadata_str)
+        return ret.format(
+            Classname=self.__class__.__name__,
+            feature=feature_str,
+            metadata=metadata_str,
+        )
 
 
 class TorchBasedFeatureStore(BasicFeatureStore):
@@ -252,6 +256,8 @@ def pin_memory_(self):
             feature.pin_memory_()
 
     def __repr__(self) -> str:
-        ret = "TorchBasedFeatureStore(\n" + "    {features}\n" + ")"
+        ret = "{Classname}(\n" + "    {features}\n" + ")"
         features_str = textwrap.indent(str(self._features), "    ").strip()
-        return ret.format(features=features_str)
+        return ret.format(
+            Classname=self.__class__.__name__, features=features_str
+        )

python/dgl/graphbolt/itemset.py

@@ -177,7 +177,7 @@ def names(self) -> Tuple[str]:
 
     def __repr__(self) -> str:
         ret = (
-            f"ItemSet(\n"
+            f"{self.__class__.__name__}(\n"
             f"    items={self._items},\n"
             f"    names={self._names},\n"
             f")"
@@ -339,7 +339,7 @@ def names(self) -> Tuple[str]:
 
     def __repr__(self) -> str:
         ret = (
-            "ItemSetDict(\n"
+            "{Classname}(\n"
             "    itemsets={itemsets},\n"
             "    names={names},\n"
             ")"
@@ -349,4 +349,8 @@ def __repr__(self) -> str:
             repr(self._itemsets), " " * len("    itemsets=")
         ).strip()
 
-        return ret.format(itemsets=itemsets_str, names=self._names)
+        return ret.format(
+            Classname=self.__class__.__name__,
+            itemsets=itemsets_str,
+            names=self._names,
+        )