Skip to content

Minor typo corrections #346

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
studyingeugene wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Minor typo corrections #346

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
810bd3f
Fix typo in comment about TorchScript support
studyingeugene Oct 31, 2025
b2e9696
Fix typo in comment about GaussianConditional.update()
studyingeugene Oct 31, 2025
6db99e6
Fix deprecation warning message in _dequantize
studyingeugene Oct 31, 2025
43ee9ed
Fix ruff lint
studyingeugene Oct 31, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions compressai/entropy_models/entropy_models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,9 @@ def dequantize(

@classmethod
def _dequantize(cls, inputs: Tensor, means: Optional[Tensor] = None) -> Tensor:
warnings.warn("_dequantize. Use dequantize instead.", stacklevel=2)
warnings.warn(
"_dequantize is deprecated. Use dequantize instead.", stacklevel=2
)
return cls.dequantize(inputs, means)

def _pmf_to_cdf(self, pmf, tail_mass, pmf_length, max_length):
Expand Down Expand Up @@ -395,7 +397,7 @@ def _get_medians(self) -> Tensor:

def update(self, force: bool = False, update_quantiles: bool = False) -> bool:
# Check if we need to update the bottleneck parameters, the offsets are
# only computed and stored when the conditonal model is update()'d.
# only computed and stored when the conditional model is update()'d.
if self._offset.numel() > 0 and not force:
return False

Expand Down Expand Up @@ -437,7 +439,7 @@ def loss(self) -> Tensor:
return loss

def _logits_cumulative(self, inputs: Tensor, stop_gradient: bool) -> Tensor:
# TorchScript not yet working (nn.Mmodule indexing not supported)
# TorchScript not yet working (nn.Module indexing not supported)
logits = inputs
for i in range(len(self.filters) + 1):
matrix = self.matrices[i]
Expand Down