From a5e88dbd2e667360150b9285aeb30d5bdb185ea0 Mon Sep 17 00:00:00 2001 From: abdulhakeem Date: Fri, 21 Feb 2025 08:54:55 -0600 Subject: [PATCH] clean up --- .../list_functions/list_functions.py | 20 +------------------ 1 file changed, 1 insertion(+), 19 deletions(-) diff --git a/reasoning_gym/induction/list_functions/list_functions.py b/reasoning_gym/induction/list_functions/list_functions.py index cf2d98e7..f91a683a 100644 --- a/reasoning_gym/induction/list_functions/list_functions.py +++ b/reasoning_gym/induction/list_functions/list_functions.py @@ -19,25 +19,7 @@ class ListFunctionsDatasetConfig: assert self.size > 0, "size must be positive" -tasks = [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, -] +tasks = list(range(17)) class ListFunctionsDataset(ProceduralDataset):