diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 1568ab1..da70278 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -3,7 +3,7 @@ default_language_version:
 
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v5.0.0
+    rev: v6.0.0
     hooks:
       # list of supported hooks: https://pre-commit.com/hooks.html
       - id: trailing-whitespace
@@ -14,12 +14,12 @@ repos:
   # python code formatting/linting
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
-    rev: "v0.12.3"
+    rev: "v0.15.0"
     hooks:
       - id: ruff
         args: [--fix]
-  - repo: https://github.com/psf/black
-    rev: 25.1.0
+  - repo: https://github.com/psf/black-pre-commit-mirror
+    rev: 26.1.0
     hooks:
       - id: black
         args: [--line-length, "100"]
diff --git a/train/run.py b/train/run.py
index 30a742e..873cfba 100644
--- a/train/run.py
+++ b/train/run.py
@@ -23,10 +23,8 @@
 def get_wandb_logger(trainer: Trainer) -> WandbLogger:
     """Return the wandb logger from the trainer."""
     if trainer.fast_dev_run:
-        raise Exception(
-            "Cannot use wandb callbacks since pytorch lightning \
-            disables loggers in `fast_dev_run=true` mode."
-        )
+        raise Exception("Cannot use wandb callbacks since pytorch lightning \
+            disables loggers in `fast_dev_run=true` mode.")
 
     if isinstance(trainer.logger, WandbLogger):
         return trainer.logger