doc

dmlc · Jun 9, 2024 · 2decac0 · 2decac0
1 parent 52638e3
commit 2decac0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/python/dgl/graphbolt/minibatch.py b/python/dgl/graphbolt/minibatch.py
@@ -163,14 +163,17 @@ def blocks(self):
         """DGL blocks extracted from `MiniBatch` containing graphical structures
         and ID mappings.
         """
+        if not self.sampled_subgraphs:
+            return None
+
         if self._blocks is None:
             self._blocks = self.compute_blocks()
         return self._blocks
 
     def compute_blocks(self):
-        if not self.sampled_subgraphs:
-            return None
-
+        """Extracts DGL blocks from `MiniBatch` to construct graphical
+        structures and ID mappings.
+        """
         is_heterogeneous = isinstance(
             self.sampled_subgraphs[0].sampled_csc, Dict
         )