Skip to content

Commit 89d5137

Browse files
YiozolmYodaEmbedding
authored andcommitted
style: satisfy static analysis
1 parent fa2e226 commit 89d5137

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

compressai/latent_codecs/channel_groups.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -84,10 +84,7 @@ def __init__(
8484
if support_slices is None:
8585
support_slices = [range(k) for k in range(len(self.groups))]
8686
assert len(support_slices) == len(self.groups)
87-
assert all(
88-
all(0 <= j < k for j in s)
89-
for k, s in enumerate(support_slices)
90-
)
87+
assert all(all(0 <= j < k for j in s) for k, s in enumerate(support_slices))
9188
self.support_slices = [tuple(support_slice) for support_slice in support_slices]
9289
self.channel_context = nn.ModuleDict(channel_context)
9390
self.latent_codec = nn.ModuleDict(latent_codec)

compressai/models/cca.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@
5252
from __future__ import annotations
5353

5454
import math
55+
5556
from itertools import accumulate
5657
from typing import Any, Dict, List, Optional, Sequence, Tuple
5758

0 commit comments

Comments
 (0)