diff --git a/pyproject.toml b/pyproject.toml
index c30243c4a7a882c0f6759727947d161bad5ca8e4..a182a7e0be6ba6f5577b46b18d7da5fee6975ef6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -275,6 +275,7 @@ exclude = [  # don't report on objects that match any of these regex
     '\.__len__$',
     '\.__getitem__$',
     '\.__iter__$',
+    '\.__exit__$',
 ]
 
 override_SS05 = [  # override SS05 to allow docstrings starting with these words
diff --git a/src/mednet/config/data/tbx11k/datamodule.py b/src/mednet/config/data/tbx11k/datamodule.py
index 9593d23d6de35d5ab874d8aaaecaf0c1cf61262c..9c76bb5af431793e2668ef4a181e4e98b28e5f25 100644
--- a/src/mednet/config/data/tbx11k/datamodule.py
+++ b/src/mednet/config/data/tbx11k/datamodule.py
@@ -118,7 +118,9 @@ class BoundingBoxes(collections.abc.Sequence[BoundingBox]):
 # We update the default collate function map to use our custom function as
 # explained at:
 # https://pytorch.org/docs/stable/data.html#torch.utils.data.default_collate
-def _collate_boundingboxes_fn(batch, *, collate_fn_map=None):
+def _collate_boundingboxes_fn(
+    batch, *, collate_fn_map=None
+):  # numpydoc ignore=PR01
     """Custom collate_fn() for pytorch dataloaders that ignores BoundingBoxes objects.
 
     Returns