Skip to content

Commit 7e54629

Browse files
authored
Fix ruff and make sure pre-commit is at same version (#1658)
stack-info: PR: #1658, branch: drisspg/stack/32
1 parent 6ffe236 commit 7e54629

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

.pre-commit-config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ repos:
1111

1212
- repo: https://github.com/astral-sh/ruff-pre-commit
1313
# Ruff version.
14-
rev: v0.5.6
14+
rev: v0.6.8
1515
hooks:
1616
# Run the linter.
1717
- id: ruff

torchao/quantization/quant_api.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -450,7 +450,9 @@ def _linear_extra_repr(self):
450450
return f"in_features={self.weight.shape[1]}, out_features={self.weight.shape[0]}, weight={_quantization_type(self.weight)}"
451451

452452

453-
def _get_linear_subclass_inserter(constructor, *, allow_requires_grad=False, propagate_bias=False, **kwargs):
453+
def _get_linear_subclass_inserter(
454+
constructor, *, allow_requires_grad=False, propagate_bias=False, **kwargs
455+
):
454456
"""Helper function to apply the constructor that quantizes the weight Tensor (with additional kwargs)
455457
to the weight of linear module
456458
"""

0 commit comments

Comments
 (0)