diff --git a/test/test_transforms.py b/test/test_transforms.py
index 32478315095..f46b5dc6d79 100644
--- a/test/test_transforms.py
+++ b/test/test_transforms.py
@@ -2284,7 +2284,9 @@ def test_batch_unlocked_with_batch_size_transformed(device):
         id="CenterCrop",
         marks=pytest.mark.skipif(not _has_tv, reason="needs torchvision dependency"),
     ),
-    pytest.param(partial(FlattenObservation, first_dim=-3, last_dim=-3), id="FlattenObservation"),
+    pytest.param(
+        partial(FlattenObservation, first_dim=-3, last_dim=-3), id="FlattenObservation"
+    ),
     pytest.param(
         partial(UnsqueezeTransform, unsqueeze_dim=-1), id="UnsqueezeTransform"
     ),