better list

rlberry-py · Oct 24, 2024 · 6f76651 · 6f76651
1 parent 7c05b98
commit 6f76651
Showing 1 changed file with 1 addition and 22 deletions.
diff --git a/rlberry/manager/tests/test_comparisons.py b/rlberry/manager/tests/test_comparisons.py
@@ -76,28 +76,7 @@ def test_compare(method, source):
             {
                 "agent": (["Agent 1"] * 10) + (["Agent 2"] * 10),
                 # "mean_eval": ([0] * 10) + ([10] * 10),
-                "mean_eval": [
-                    1,
-                    2,
-                    3,
-                    4,
-                    5,
-                    6,
-                    7,
-                    8,
-                    9,
-                    10,
-                    11,
-                    12,
-                    13,
-                    14,
-                    15,
-                    16,
-                    17,
-                    18,
-                    19,
-                    20,
-                ],
+                "mean_eval": [range(12, 21)],
             }
         )