Skip to content

Commit 3888f06

Browse files
committed
fixup! [feat] Use ruff for formatting and linting new files by default
Signed-off-by: Robin Kobus <[email protected]>
1 parent 7871da6 commit 3888f06

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

.pre-commit-config.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -267,6 +267,7 @@ common-files: &common_files |
267267
tensorrt_llm/_torch/attention_backend/flashinfer.py |
268268
tensorrt_llm/_torch/attention_backend/interface.py |
269269
tensorrt_llm/_torch/attention_backend/sparse/__init__.py |
270+
tensorrt_llm/_torch/attention_backend/sparse/dsa.py |
270271
tensorrt_llm/_torch/attention_backend/sparse/kernel.py |
271272
tensorrt_llm/_torch/attention_backend/sparse/rocket.py |
272273
tensorrt_llm/_torch/attention_backend/sparse/utils.py |
@@ -286,6 +287,7 @@ common-files: &common_files |
286287
tensorrt_llm/_torch/compilation/recover_pass.py |
287288
tensorrt_llm/_torch/compilation/remove_copy_pass.py |
288289
tensorrt_llm/_torch/compilation/utils.py |
290+
tensorrt_llm/_torch/configs/deepseek_v3.py |
289291
tensorrt_llm/_torch/cublaslt_utils.py |
290292
tensorrt_llm/_torch/custom_ops/__init__.py |
291293
tensorrt_llm/_torch/custom_ops/cpp_custom_ops.py |
@@ -582,6 +584,7 @@ common-files: &common_files |
582584
tensorrt_llm/functional.py |
583585
tensorrt_llm/inputs/__init__.py |
584586
tensorrt_llm/inputs/data.py |
587+
tensorrt_llm/inputs/evs.py |
585588
tensorrt_llm/inputs/multimodal.py |
586589
tensorrt_llm/inputs/registry.py |
587590
tensorrt_llm/inputs/utils.py |
@@ -973,7 +976,10 @@ common-files: &common_files |
973976
tests/scripts/iteration_log_parser.py |
974977
tests/scripts/perf-sanity/parse_benchmark_results.py |
975978
tests/scripts/perf-sanity/run_benchmark_serve.py |
979+
tests/unittest/_torch/attention/sparse/test_dsa_indexer.py |
980+
tests/unittest/_torch/attention/sparse/test_flash_mla.py |
976981
tests/unittest/_torch/attention/sparse/test_rocketkv.py |
982+
tests/unittest/_torch/attention/sparse/test_sparse_mla_forward.py |
977983
tests/unittest/_torch/attention/test_attention_mla.py |
978984
tests/unittest/_torch/attention/test_attention_no_cache.py |
979985
tests/unittest/_torch/attention/test_attention.py |
@@ -1040,8 +1046,10 @@ common-files: &common_files |
10401046
tests/unittest/_torch/multimodal/test_share_multiparams.py |
10411047
tests/unittest/_torch/pattern_watcher.py |
10421048
tests/unittest/_torch/ray_orchestrator/conftest.py |
1049+
tests/unittest/_torch/ray_orchestrator/multi_gpu/test_executor.py |
10431050
tests/unittest/_torch/ray_orchestrator/multi_gpu/test_mapping.py |
10441051
tests/unittest/_torch/ray_orchestrator/multi_gpu/test_ops_ray.py |
1052+
tests/unittest/_torch/ray_orchestrator/multi_gpu/test_ops.py |
10451053
tests/unittest/_torch/ray_orchestrator/multi_gpu/test_placement.py |
10461054
tests/unittest/_torch/ray_orchestrator/single_gpu/test_cache_transceiver_comm.py |
10471055
tests/unittest/_torch/sampler/test_beam_search.py |

pyproject.toml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -307,6 +307,7 @@ exclude = [
307307
"tensorrt_llm/_torch/attention_backend/flashinfer.py",
308308
"tensorrt_llm/_torch/attention_backend/interface.py",
309309
"tensorrt_llm/_torch/attention_backend/sparse/__init__.py",
310+
"tensorrt_llm/_torch/attention_backend/sparse/dsa.py",
310311
"tensorrt_llm/_torch/attention_backend/sparse/kernel.py",
311312
"tensorrt_llm/_torch/attention_backend/sparse/rocket.py",
312313
"tensorrt_llm/_torch/attention_backend/sparse/utils.py",
@@ -326,6 +327,7 @@ exclude = [
326327
"tensorrt_llm/_torch/compilation/recover_pass.py",
327328
"tensorrt_llm/_torch/compilation/remove_copy_pass.py",
328329
"tensorrt_llm/_torch/compilation/utils.py",
330+
"tensorrt_llm/_torch/configs/deepseek_v3.py",
329331
"tensorrt_llm/_torch/cublaslt_utils.py",
330332
"tensorrt_llm/_torch/custom_ops/__init__.py",
331333
"tensorrt_llm/_torch/custom_ops/cpp_custom_ops.py",
@@ -622,6 +624,7 @@ exclude = [
622624
"tensorrt_llm/functional.py",
623625
"tensorrt_llm/inputs/__init__.py",
624626
"tensorrt_llm/inputs/data.py",
627+
"tensorrt_llm/inputs/evs.py",
625628
"tensorrt_llm/inputs/multimodal.py",
626629
"tensorrt_llm/inputs/registry.py",
627630
"tensorrt_llm/inputs/utils.py",
@@ -1013,7 +1016,10 @@ exclude = [
10131016
"tests/scripts/iteration_log_parser.py",
10141017
"tests/scripts/perf-sanity/parse_benchmark_results.py",
10151018
"tests/scripts/perf-sanity/run_benchmark_serve.py",
1019+
"tests/unittest/_torch/attention/sparse/test_dsa_indexer.py",
1020+
"tests/unittest/_torch/attention/sparse/test_flash_mla.py",
10161021
"tests/unittest/_torch/attention/sparse/test_rocketkv.py",
1022+
"tests/unittest/_torch/attention/sparse/test_sparse_mla_forward.py",
10171023
"tests/unittest/_torch/attention/test_attention_mla.py",
10181024
"tests/unittest/_torch/attention/test_attention_no_cache.py",
10191025
"tests/unittest/_torch/attention/test_attention.py",
@@ -1080,8 +1086,10 @@ exclude = [
10801086
"tests/unittest/_torch/multimodal/test_share_multiparams.py",
10811087
"tests/unittest/_torch/pattern_watcher.py",
10821088
"tests/unittest/_torch/ray_orchestrator/conftest.py",
1089+
"tests/unittest/_torch/ray_orchestrator/multi_gpu/test_executor.py",
10831090
"tests/unittest/_torch/ray_orchestrator/multi_gpu/test_mapping.py",
10841091
"tests/unittest/_torch/ray_orchestrator/multi_gpu/test_ops_ray.py",
1092+
"tests/unittest/_torch/ray_orchestrator/multi_gpu/test_ops.py",
10851093
"tests/unittest/_torch/ray_orchestrator/multi_gpu/test_placement.py",
10861094
"tests/unittest/_torch/ray_orchestrator/single_gpu/test_cache_transceiver_comm.py",
10871095
"tests/unittest/_torch/sampler/test_beam_search.py",
@@ -1426,6 +1434,7 @@ exclude = [
14261434
"triton_backend/tools/whisper/client.py"
14271435
]
14281436

1437+
14291438
[tool.ruff.format]
14301439
# Like Black, respect magic trailing commas.
14311440
skip-magic-trailing-comma = false

0 commit comments

Comments
 (0)