fixes

ServiceNow · Jan 22, 2025 · 77b1324 · 77b1324
1 parent 8245041
commit 77b1324
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
@@ -270,20 +270,21 @@ def test_gpt_concatenate_data():
 
 
 GPT_SLICE_EXPECTED_TRAINING_SAMPLES = [
-    [1725, 74, 207, 1635, 4440, 2774],
-    [359, 489, 4266, 2052, 5351, 80],
-    [374, 7534, 87, 1073, 79, 480],
-    [8008, 498, 71, 727, 80, 315],
+    [2625, 76, 2625, 2639, 74, 243],
+    [207, 481, 5546, 74, 414, 498],
+    [74, 333, 1963, 310, 5337, 3628],
+    [79, 2361, 80, 2012, 84, 480],
 ]
+
 GPT_SLICE_EXPECTED_VALIDATION_SAMPLES = [
-    [1725, 74, 207, 1635, 4440, 2774],
-    [359, 489, 4266, 2052, 5351, 80],
-    [374, 7534, 87, 1073, 79, 480],
-    [8008, 498, 71, 727, 80, 315],
-    [2210, 8179, 73, 2582, 897, 1178],
-    [409, 5091, 328, 1378, 5483, 88],
-    [83, 4457, 3316, 333, 489, 317],
-    [330, 155, 2449, 1136, 1106, 5370],
+    [2352, 3687, 2311, 4900, 542, 3732],
+    [2551, 5283, 900, 3140, 328, 68],
+    [7979, 2283, 329, 727, 2740, 2818],
+    [4117, 8056, 79, 1798, 243, 498],
+    [243, 542, 387, 6476, 6686, 785],
+    [95, 6641, 207, 279, 2304, 602],
+    [89, 4446, 947, 293, 947, 1544],
+    [243, 3712, 86, 476, 80, 2547],
 ]