Skip to content

Commit 4fbc02f

Browse files
studyingeugenefracape
authored andcommitted
refactor: fix lint errors
Fix lint errors in entropy_models.py
1 parent 32f4339 commit 4fbc02f

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

compressai/entropy_models/entropy_models.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,7 @@ def __init__(self, method):
5454

5555
if method not in available_entropy_coders():
5656
methods = ", ".join(available_entropy_coders())
57-
raise ValueError(
58-
f'Unknown entropy coder "{method}"' f" (available: {methods})"
59-
)
57+
raise ValueError(f'Unknown entropy coder "{method}" (available: {methods})')
6058

6159
if method == "ans":
6260
from compressai import ans
@@ -474,7 +472,7 @@ def forward(
474472
if training is None:
475473
training = self.training
476474

477-
D = x.dim()
475+
D = x.dim()
478476
# B C ... -> C B ...
479477
perm = [1, 0] + list(range(2, D))
480478
inv_perm = [0] * D

0 commit comments

Comments
 (0)