Skip to content

Commit dd9a053

Browse files
authored
solve et lint-mypy issue (pytorch#18587) (pytorch#18587)
Summary: try to solve lint-mypy issue in torch.hud Differential Revision: D98762153
1 parent 15e8bf7 commit dd9a053

4 files changed

Lines changed: 8 additions & 8 deletions

File tree

backends/arm/test/models/Gemma3n/gemma3n_test_config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def get_gemma3n_text_test_config() -> Gemma3nTextConfig:
2323
altup_num_inputs=4,
2424
altup_active_idx=0,
2525
altup_correct_scale=True,
26-
)
26+
) # type: ignore[call-arg]
2727
# Force eager attention path to keep the module exportable in tests.
2828
config._attn_implementation = "eager"
2929
return config
@@ -34,4 +34,4 @@ def get_gemma3n_audio_test_config() -> Gemma3nAudioConfig:
3434
input_feat_size=128,
3535
hidden_size=256,
3636
rms_norm_eps=1e-6,
37-
)
37+
) # type: ignore[call-arg]

backends/arm/test/models/phi3/phi3_module_test_configs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def get_phi3_test_config() -> Phi3Config:
1818
original_max_position_embeddings=32,
1919
use_cache=False,
2020
tie_word_embeddings=False,
21-
)
21+
) # type: ignore[call-arg]
2222
# Force eager attention path to keep the module exportable in tests.
2323
config._attn_implementation = "eager"
2424
return config

backends/arm/test/models/stable_diffusion/stable_diffusion_module_test_configs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434

3535

3636
# Source: https://github.com/huggingface/diffusers/blob/v0.33.1/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3.py#L56
37-
CLIP_text_encoder_config = CLIPTextConfig(
37+
CLIP_text_encoder_config = CLIPTextConfig( # type: ignore[call-arg]
3838
bos_token_id=0,
3939
eos_token_id=2,
4040
hidden_size=32,
@@ -51,7 +51,7 @@
5151

5252
# Source: https://github.com/huggingface/diffusers/blob/v0.33.1/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3.py#L76
5353
# Exported from: T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5").config
54-
T5_encoder_config = T5Config(
54+
T5_encoder_config = T5Config( # type: ignore[call-arg]
5555
bos_token_id=0,
5656
classifier_dropout=0.0,
5757
d_ff=37,

backends/arm/test/modules/test_static_cache.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,11 @@
2828

2929

3030
test_configs = {
31-
"multihead_attention": LlamaConfig(num_attention_heads=32),
32-
"grouped_query_attention": LlamaConfig(
31+
"multihead_attention": LlamaConfig(num_attention_heads=32), # type: ignore[call-arg]
32+
"grouped_query_attention": LlamaConfig( # type: ignore[call-arg]
3333
num_attention_heads=32, num_key_value_heads=4
3434
),
35-
"multi_query_attention": LlamaConfig(num_attention_heads=32, num_key_value_heads=1),
35+
"multi_query_attention": LlamaConfig(num_attention_heads=32, num_key_value_heads=1), # type: ignore[call-arg]
3636
}
3737

3838

0 commit comments

Comments
 (0)