diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 307f8b3..42819bf 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -10,7 +10,7 @@ repos:
     -   id: check-added-large-files
 
 -   repo: https://github.com/psf/black
-    rev: 24.10.0
+    rev: 25.1.0
     hooks:
     -   id: black
 
diff --git a/fairseq/modules/checkpoint_activations.py b/fairseq/modules/checkpoint_activations.py
index 0717271..3ad7330 100644
--- a/fairseq/modules/checkpoint_activations.py
+++ b/fairseq/modules/checkpoint_activations.py
@@ -99,7 +99,7 @@ def unpack_kwargs(
 
 
 def split_non_tensors(
-    mixed: Union[torch.Tensor, Tuple[Any]]
+    mixed: Union[torch.Tensor, Tuple[Any]],
 ) -> Tuple[Tuple[torch.Tensor], Dict[str, List[Any]]]:
     """
     Usage::