From f6e86bd0a64b48de9cb7dbcd4e70f8194f43272b Mon Sep 17 00:00:00 2001
From: Finbarr Timbers <finbarrtimbers@gmail.com>
Date: Sun, 3 Nov 2024 13:19:39 -0700
Subject: [PATCH] Ran ruff.

---
 llamax/integration_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llamax/integration_test.py b/llamax/integration_test.py
index 82d4f52..a80ab14 100644
--- a/llamax/integration_test.py
+++ b/llamax/integration_test.py
@@ -125,7 +125,7 @@ def setUpClass(cls):
 
             # Convert checkpoint to double precision
             checkpoint = {k: v.double() for k, v in checkpoint.items()}
-            
+
             jax.tree.map(
                 lambda x, y: np.testing.assert_array_equal(x.shape, y.shape),
                 dict(cls.torch_model.state_dict()),