[GraphBolt] Move `get_attributes` function to internal. (#6634)

Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>

python/dgl/graphbolt/impl/sampled_subgraph_impl.py

@@ -6,6 +6,7 @@ from typing import Dict, Tuple, Union
 import torch
 
 from ..base import CSCFormatBase, etype_str_to_tuple
+from ..internal import get_attributes
 from ..sampled_subgraph import SampledSubgraph
 
 __all__ = ["FusedSampledSubgraphImpl", "SampledSubgraphImpl"]
@@ -140,16 +141,7 @@ class SampledSubgraphImpl(SampledSubgraph):
 def _sampled_subgraph_str(sampled_subgraph: SampledSubgraph, classname) -> str:
     final_str = classname + "("
 
-    def _get_attributes(_obj) -> list:
-        attributes = [
-            attribute
-            for attribute in dir(_obj)
-            if not attribute.startswith("__")
-            and not callable(getattr(_obj, attribute))
-        ]
-        return attributes
-
-    attributes = _get_attributes(sampled_subgraph)
+    attributes = get_attributes(sampled_subgraph)
     attributes.reverse()
 
     for name in attributes:

python/dgl/graphbolt/internal/utils.py

@@ -109,3 +109,14 @@ def copy_or_convert_data(
         if is_feature and data.dim() == 1:
             data = data.reshape(-1, 1)
         save_data(data, output_path, output_format)
+
+
+def get_attributes(_obj) -> list:
+    """Get attributes of the class."""
+    attributes = [
+        attribute
+        for attribute in dir(_obj)
+        if not attribute.startswith("__")
+        and not callable(getattr(_obj, attribute))
+    ]
+    return attributes

python/dgl/graphbolt/minibatch.py

@@ -10,6 +10,7 @@ from dgl.heterograph import DGLBlock
 from dgl.utils import recursive_apply
 
 from .base import CSCFormatBase, etype_str_to_tuple
+from .internal import get_attributes
 from .sampled_subgraph import SampledSubgraph
 
 __all__ = ["DGLMiniBatch", "MiniBatch"]
@@ -574,17 +575,7 @@ class MiniBatch:
 def _minibatch_str(minibatch: MiniBatch) -> str:
     final_str = ""
     # Get all attributes in the class except methods.
-
-    def _get_attributes(_obj) -> list:
-        attributes = [
-            attribute
-            for attribute in dir(_obj)
-            if not attribute.startswith("__")
-            and not callable(getattr(_obj, attribute))
-        ]
-        return attributes
-
-    attributes = _get_attributes(minibatch)
+    attributes = get_attributes(minibatch)
     attributes.reverse()
     # Insert key with its value into the string.
     for name in attributes:
@@ -617,17 +608,7 @@ def _minibatch_str(minibatch: MiniBatch) -> str:
 def _dgl_minibatch_str(dglminibatch: DGLMiniBatch) -> str:
     final_str = ""
     # Get all attributes in the class except methods.
-
-    def _get_attributes(_obj) -> list:
-        attributes = [
-            attribute
-            for attribute in dir(_obj)
-            if not attribute.startswith("__")
-            and not callable(getattr(_obj, attribute))
-        ]
-        return attributes
-
-    attributes = _get_attributes(dglminibatch)
+    attributes = get_attributes(dglminibatch)
     attributes.reverse()
     # Insert key with its value into the string.
     for name in attributes: