From e35b4622cc82cd0c774ee009893fa29371a1a2cb Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Sun, 3 Nov 2024 07:30:50 +0000
Subject: [PATCH] Fix granite unit test

---
 tests/tiny_random.test.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/tiny_random.test.js b/tests/tiny_random.test.js
index 33413ecfd..e0a3d9369 100644
--- a/tests/tiny_random.test.js
+++ b/tests/tiny_random.test.js
@@ -1108,7 +1108,7 @@ describe("Tiny random models", () => {
             ...inputs,
             max_length: 10,
           });
-          expect(outputs.tolist()).toEqual([[7656n, 23147n, 31291n, 1011n, 8768n, 30904n, 9256n, 28368n, 16199n, 26560n]]);
+          expect(outputs.tolist()).toEqual([[7656n, 39727n, 33077n, 9643n, 30539n, 47869n, 48739n, 15085n, 9203n, 14020n]]);
         },
         MAX_TEST_EXECUTION_TIME,
       );
@@ -1122,8 +1122,8 @@ describe("Tiny random models", () => {
             max_length: 10,
           });
           expect(outputs.tolist()).toEqual([
-            [0n, 7656n, 23147n, 31291n, 1011n, 8768n, 30904n, 9256n, 28368n, 16199n],
-            [7656n, 5788n, 9477n, 14490n, 18374n, 28650n, 10907n, 2989n, 14096n, 27403n],
+            [0n, 7656n, 39727n, 33077n, 9643n, 30539n, 47869n, 48739n, 15085n, 9203n],
+            [7656n, 5788n, 17835n, 13234n, 7592n, 21471n, 30537n, 23023n, 43450n, 4824n],
           ]);
         },
         MAX_TEST_EXECUTION_TIME,