From 3366677ae23b18cfe7a322371862c7c344d1efc2 Mon Sep 17 00:00:00 2001 From: vmoens Date: Tue, 10 Jan 2023 10:05:25 +0000 Subject: [PATCH] lint --- test/test_transforms.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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" ),