From 69af7793d3ff58733dd0ff7655c42539ecebc257 Mon Sep 17 00:00:00 2001
From: alexhernandezgarcia <alexhdezgarcia@gmail.com>
Date: Thu, 8 Feb 2024 18:11:18 -0500
Subject: [PATCH] Add type of return for select()

---
 gflownet/utils/buffer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gflownet/utils/buffer.py b/gflownet/utils/buffer.py
index 024d3a52a..bc89658e2 100644
--- 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.