Skip to content

Commit 64dabb5

Browse files
committed
style: silence F821 on calls to removed transform_with_custom_compression_ops
The 8 call sites are in @pytest.mark.skip'd tests for a removed API; add # noqa: F821 so ruff passes until the obsolete tests are rewritten or removed.
1 parent ea899b4 commit 64dabb5

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

tests/test_externalize.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3065,7 +3065,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
30653065
)
30663066
quantizer = PostTrainingQuantizer(model, quantization_config)
30673067
model = cast("nn.Module", quantizer.compress())
3068-
transform_with_custom_compression_ops(model)
3068+
transform_with_custom_compression_ops(model) # noqa: F821
30693069

30703070
sample = (torch.randn(2, 24),)
30713071

@@ -3153,7 +3153,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
31533153
)
31543154
quantizer = PostTrainingQuantizer(model, quantization_config)
31553155
model = cast("nn.Module", quantizer.compress())
3156-
transform_with_custom_compression_ops(model)
3156+
transform_with_custom_compression_ops(model) # noqa: F821
31573157

31583158
sample = (torch.randn(2, 24),)
31593159

@@ -3217,7 +3217,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
32173217
)
32183218
quantizer = PostTrainingQuantizer(model, quantization_config)
32193219
model = cast("nn.Module", quantizer.compress())
3220-
transform_with_custom_compression_ops(model)
3220+
transform_with_custom_compression_ops(model) # noqa: F821
32213221

32223222
x = torch.randn(2, 1, 1, in_dim)
32233223
indices = torch.tensor([[0, 2], [1, 3]], dtype=torch.int16)
@@ -3318,7 +3318,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
33183318
)
33193319
quantizer = PostTrainingQuantizer(model, quantization_config)
33203320
model = cast("nn.Module", quantizer.compress())
3321-
transform_with_custom_compression_ops(model)
3321+
transform_with_custom_compression_ops(model) # noqa: F821
33223322

33233323
x = torch.randn(2, 1, 1, in_dim)
33243324
indices = torch.tensor([[0, 2], [1, 3]], dtype=torch.int16)
@@ -3390,7 +3390,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
33903390
)
33913391
quantizer = PostTrainingQuantizer(model, quantization_config)
33923392
model = cast("nn.Module", quantizer.compress())
3393-
transform_with_custom_compression_ops(model)
3393+
transform_with_custom_compression_ops(model) # noqa: F821
33943394

33953395
sample = (torch.randn(1, 4, embed_dim),)
33963396

@@ -3484,7 +3484,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
34843484
)
34853485
quantizer = PostTrainingQuantizer(model, quantization_config)
34863486
model = cast("nn.Module", quantizer.compress())
3487-
transform_with_custom_compression_ops(model)
3487+
transform_with_custom_compression_ops(model) # noqa: F821
34883488

34893489
sample = (torch.randn(1, 4, embed_dim),)
34903490

@@ -3553,7 +3553,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
35533553
)
35543554
quantizer = PostTrainingQuantizer(model, quantization_config)
35553555
model = cast("nn.Module", quantizer.compress())
3556-
transform_with_custom_compression_ops(model)
3556+
transform_with_custom_compression_ops(model) # noqa: F821
35573557

35583558
x = torch.randn(2, in_dim)
35593559
indices = torch.tensor([[0, 2], [1, 3]], dtype=torch.int16)
@@ -3637,7 +3637,7 @@ def forward(self, x: torch.Tensor, indices: torch.Tensor) -> torch.Tensor:
36373637
)
36383638
quantizer = PostTrainingQuantizer(model, quantization_config)
36393639
model = cast("nn.Module", quantizer.compress())
3640-
transform_with_custom_compression_ops(model)
3640+
transform_with_custom_compression_ops(model) # noqa: F821
36413641

36423642
x = torch.randn(2, in_dim)
36433643
indices = torch.tensor([[0, 2], [1, 3]], dtype=torch.int16)

0 commit comments

Comments
 (0)