Skip to content

Commit 43c6221

Browse files
committed
lint
1 parent eee8889 commit 43c6221

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

test/test_encoders.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -804,9 +804,13 @@ def test_contiguity(self, method, tmp_path, device):
804804
# frame tensors, one is contiguous while the other is non-contiguous.
805805

806806
num_frames, channels, height, width = 5, 3, 256, 256
807-
contiguous_frames = torch.randint(
808-
0, 256, size=(num_frames, channels, height, width), dtype=torch.uint8
809-
).contiguous().to(device)
807+
contiguous_frames = (
808+
torch.randint(
809+
0, 256, size=(num_frames, channels, height, width), dtype=torch.uint8
810+
)
811+
.contiguous()
812+
.to(device)
813+
)
810814
assert contiguous_frames.is_contiguous()
811815

812816
# Permute NCHW to NHWC, then update the memory layout, then permute back

0 commit comments

Comments
 (0)