diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index fe43f63..abeb646 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -10,7 +10,7 @@ repos: - id: check-added-large-files - repo: https://github.com/psf/black-pre-commit-mirror - rev: 25.12.0 + rev: 26.1.0 hooks: - id: black diff --git a/fairseq/models/__init__.py b/fairseq/models/__init__.py index 7689230..999ff93 100644 --- a/fairseq/models/__init__.py +++ b/fairseq/models/__init__.py @@ -26,7 +26,6 @@ FairseqMultiModel, ) - MODEL_REGISTRY = {} MODEL_DATACLASS_REGISTRY = {} ARCH_MODEL_REGISTRY = {} diff --git a/fairseq/models/wav2vec/wav2vec2.py b/fairseq/models/wav2vec/wav2vec2.py index 2327a8d..6c557e3 100644 --- a/fairseq/models/wav2vec/wav2vec2.py +++ b/fairseq/models/wav2vec/wav2vec2.py @@ -890,7 +890,7 @@ def make_conv(): self.conv_layers = nn.ModuleList() for i, cl in enumerate(conv_layers): assert len(cl) == 3, "invalid conv definition: " + str(cl) - (dim, k, stride) = cl + dim, k, stride = cl self.conv_layers.append( block( diff --git a/fairseq/search.py b/fairseq/search.py index 25161d6..2e812eb 100644 --- a/fairseq/search.py +++ b/fairseq/search.py @@ -879,7 +879,7 @@ def step( final_scores = torch.empty(0).to(lprobs) final_indices = torch.LongTensor().to(device=lprobs.device) final_beams = torch.LongTensor().to(device=lprobs.device) - (final_scores, final_indices) = torch.topk( + final_scores, final_indices = torch.topk( torch.stack(s_list, dim=1).view(bsz, -1), k, ) diff --git a/fairseq/tasks/__init__.py b/fairseq/tasks/__init__.py index a15abe3..1c16eb4 100644 --- a/fairseq/tasks/__init__.py +++ b/fairseq/tasks/__init__.py @@ -14,7 +14,6 @@ from .fairseq_task import FairseqTask # noqa - # register dataclass TASK_DATACLASS_REGISTRY = {} TASK_REGISTRY = {}