Add type of return for select()

alexhernandezgarcia · Feb 8, 2024 · 69af779 · 69af779
1 parent 9c18890
commit 69af779
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gflownet/utils/buffer.py b/gflownet/utils/buffer.py
@@ -276,7 +276,7 @@ def select(
         n: int,
         mode: str = "permutation",
         rng: np.random.Generator = None,
-    ):
+    ) -> List:
         """
         Selects a subset of n data points from data_dict, according to the criterion
         indicated by mode.