Skip to content

Commit 36da6de

Browse files
committed
fix: ruff formatting and PT006 parametrize tuple fix
Signed-off-by: Hung-Yueh Chiang <hungyuehc@nvidia.com>
1 parent a7d1170 commit 36da6de

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/gpu_megatron/torch/export/test_mcore_qwen3vl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def test_layernorms_are_replicated(self, key):
142142
assert mapping.func_kwargs == REPLICATE
143143

144144
@pytest.mark.parametrize(
145-
"key,expected_kwargs",
145+
("key", "expected_kwargs"),
146146
[
147147
("word_embeddings", COL_TP),
148148
("output_layer", COL_TP),

0 commit comments

Comments
 (0)