Add return to docstring

alexhernandezgarcia · Feb 9, 2024 · b9ffd74 · b9ffd74
1 parent 69af779
commit b9ffd74
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/gflownet/utils/buffer.py b/gflownet/utils/buffer.py
@@ -303,6 +303,11 @@ def select(
         rng : np.random.Generator
             A numpy random number generator, used for the permutation mode. Ignored
             otherwise.
+
+        Returns
+        -------
+        list
+            A batch of n samples, selected from data_dict.
         """
         if n == 0:
             return []