Ran ruff.

finbarrtimbers · Nov 3, 2024 · f6e86bd · f6e86bd
1 parent bef6ef2
commit f6e86bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()),