diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 0492f4c..781f5bc 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,13 +1,13 @@
 repos:
 -   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v5.0.0
+    rev: v6.0.0
     hooks:
     -   id: check-yaml
     -   id: end-of-file-fixer
     -   id: trailing-whitespace
 - repo: https://github.com/astral-sh/ruff-pre-commit
   # Ruff version.
-  rev: v0.9.9
+  rev: v0.15.7
   hooks:
     # Run the linter.
     - id: ruff
diff --git a/tests/core/test_loss.py b/tests/core/test_loss.py
index 0496743..abc74c6 100644
--- a/tests/core/test_loss.py
+++ b/tests/core/test_loss.py
@@ -452,8 +452,8 @@ def test_dataset_item_mismatched_lengths(self):
         eval_dataset_mask_shorter = [item_mask_shorter]
         eval_dataset_ids_shorter = [item_ids_shorter]
 
-        self.mock_tokenizer.decode.side_effect = (
-            lambda ids, skip_special_tokens: "a" * len(ids)
+        self.mock_tokenizer.decode.side_effect = lambda ids, skip_special_tokens: (
+            "a" * len(ids)
         )
 
         # Scenario 1: target_mask is shorter
@@ -468,8 +468,8 @@ def test_dataset_item_mismatched_lengths(self):
         self.mock_tokenizer.decode.assert_called_with([2, 3], skip_special_tokens=True)
 
         self.mock_tokenizer.reset_mock()  # Reset mock for the next scenario
-        self.mock_tokenizer.decode.side_effect = (
-            lambda ids, skip_special_tokens: "b" * len(ids)
+        self.mock_tokenizer.decode.side_effect = lambda ids, skip_special_tokens: (
+            "b" * len(ids)
         )
 
         # Scenario 2: input_ids is shorter