fix(tests): Resolve late binding of loop variable in assert message lambda

The `msg` parameter for `torch.testing.assert_close` within the `test_mamba_chunk_scan_cont_batch_prefill_chunking` test function used a lambda defined inside a for-loop.
This lambda captured the loop variable `i`, triggering the `B023` warning from the Ruff linter. Ruff implements this rule (originally from flake8-bugbear) to detect "late binding" issues, where a lambda captures a reference to a variable, not its value at the time of definition.
Although the current test runner likely executes the lambda immediately upon assertion failure, this pattern is a latent bug. Future changes could defer message generation, causing all failure messages to incorrectly display the final value of `i`, which would be misleading for debugging.
This commit fixes the issue by using a default argument in the lambda (`lambda x, i=i: ...`) to capture the value of `i` at definition time.
This robustly resolves the potential bug and allows for the removal of the `# noqa: B023` suppressions.

Signed-off-by: lyd1992 <liuyudong@iscas.ac.cn>
Signed-off-by: ihb2032 <1355790728@qq.com

Signed-off-by: ihb2032 <1355790728@qq.com>

Author: ihb2032

hq

@@ -0,0 +1,158 @@
+  origin/HEAD -> origin/main
+  origin/feat/riscv-supported-dtypes
+  origin/fix/cpu-riscv-compatibility
+  origin/fix/riscv-scalar-fp32-workaround
+  origin/main
+  upstream/7snzwi-codex/change-default-logging-behavior
+  upstream/acc-rate
+  upstream/add-nixl-transfer-time-logging
+  upstream/add-utils
+  upstream/amd-ci
+  upstream/amd_dev
+  upstream/amd_mori
+  upstream/avoid-double-free
+  upstream/bench-latency
+  upstream/benchmark
+  upstream/benchmark-output
+  upstream/benchmark_serving_test
+  upstream/bind_kv_caches
+  upstream/build-flashinfer-aot-wheel
+  upstream/codex/add-auto-max-model-length-setting
+  upstream/codex/add-pandas-and-datasets-to-requirements
+  upstream/codex/change-default-logging-behavior
+  upstream/codex/fix-torch_cuda_arch_list-in-dockerfile
+  upstream/codex/remove-raydistributedexecutor-from-v0-engine
+  upstream/codex/update-arch-overview-md-with-vllm-v1-details
+  upstream/compile-eplb
+  upstream/copilot/fix-31e676e9-a4af-4ed2-b74d-19d27f0a57b2
+  upstream/copilot/fix-56244f30-e76a-41ed-beaf-3bc9de22a2c9
+  upstream/copilot/fix-584be906-f283-4e17-8776-c14111357ee7
+  upstream/copilot/fix-870996da-9146-438e-9a52-cdc6c1743086
+  upstream/copilot/fix-c6914add-1b66-46d0-9948-c2e7b6f2259f
+  upstream/copilot/fix-cudagraph-flag-combination
+  upstream/correct-docs-cuda-version
+  upstream/dbo-cudagraph-size-cherry
+  upstream/debug
+  upstream/debug-logging
+  upstream/debug-logs
+  upstream/decode_bench_connector
+  upstream/deep_full_cudagraph_fix
+  upstream/deepep_tweaks
+  upstream/dependabot/github_actions/actions/checkout-5.0.0
+  upstream/dependabot/github_actions/actions/setup-python-5.6.0
+  upstream/disable-sd
+  upstream/dockerfile-nvcc-compress
+  upstream/dynamo-patch
+  upstream/fix-doc-build
+  upstream/fix-hashing-partial-blocks
+  upstream/fix-precommit
+  upstream/fix-v1-test
+  upstream/fix_ds_eagle
+  upstream/fix_hang
+  upstream/fix_use_ep
+  upstream/fp8_ep_dp
+  upstream/full_cudagraph
+  upstream/fused-moe-tuning-ep
+  upstream/gemma3n-mm
+  upstream/gpu-ids
+  upstream/gpu_ids2
+  upstream/il_tool
+  upstream/jax-tpu
+  upstream/khluu/clean_apt
+  upstream/khluu/nccl
+  upstream/khluu/test
+  upstream/khluu/test_fixed_premerge
+  upstream/khluu/test_latest_feat
+  upstream/khluu/test_us_east_1
+  upstream/khluu/try_moc
+  upstream/khluu/use_ccache_premerge
+  upstream/low_latency_opt
+  upstream/lwilkinson/cg-support
+  upstream/lwilkinson/dbo-full-cudagraphs
+  upstream/lwilkinson/eagle-piecewise
+  upstream/lwilkinson/potential-cutlass-mla-fix
+  upstream/lwilkinson/refactor-cmake
+  upstream/main
+  upstream/mamba-prefix-cache-postmerge
+  upstream/mamba_tests
+  upstream/marlin_gptoss_swiglu
+  upstream/maybe_fix_hang_2
+  upstream/memory-leak-branch
+  upstream/mergify/houseroad/config-update
+  upstream/minus_x
+  upstream/mla-support-awq-marlin
+  upstream/mla_cuda_graphs
+  upstream/mla_decode_any_head
+  upstream/moondream2
+  upstream/nixl-debug-oh-fixed
+  upstream/nixl-upstreaming
+  upstream/optimize-prefix-caching-scheduling
+  upstream/pd_scheduling
+  upstream/pil_image
+  upstream/prune-samplers-test
+  upstream/qwen25vl
+  upstream/reduce_scatter_comm
+  upstream/releases/v0.10.0
+  upstream/releases/v0.10.1
+  upstream/releases/v0.10.2
+  upstream/releases/v0.11.0
+  upstream/releases/v0.9.0
+  upstream/releases/v0.9.1
+  upstream/releases/v0.9.2
+  upstream/remove-async-engine-tests
+  upstream/remove-metrics-and-tracing-test
+  upstream/remove-regression-test
+  upstream/remove_mamba_ssm
+  upstream/revert-21550-chengji/fix-ci
+  upstream/revert-22299-main
+  upstream/rob-fixes
+  upstream/running-deque
+  upstream/sampler-env-variable
+  upstream/seemethere/cuda_arm64
+  upstream/simon-mo-patch-1
+  upstream/skip-lmfe-tests
+  upstream/split_kv_cache_init
+  upstream/support_global_dp_logging
+  upstream/test-debug-lb
+  upstream/tms/distributed_timeout
+  upstream/topk_id_hack
+  upstream/torch-2.8
+  upstream/torch_dynamo
+  upstream/tpu_v1
+  upstream/tpu_v1_optimized
+  upstream/triton-configs
+  upstream/use-uv-python-for-docker
+  upstream/v0.7.2-staging-branch
+  upstream/v0.8.0
+  upstream/v0.8.1
+  upstream/v0.8.2
+  upstream/v0.8.3
+  upstream/v0.8.4
+  upstream/v0.8.5
+  upstream/v1-block-table-opt
+  upstream/v1-blocktable-opt
+  upstream/v1-sched-interface-2
+  upstream/v1_fix_profiler
+  upstream/wentao-enable-e8m0-by-default
+  upstream/wentao-fix-negative-cuda-memory-usage
+  upstream/wentao-fix-test-in-batch-invariance
+  upstream/wentao-optimize-log
+  upstream/wentao-refactor-fp8-moe-backend-choice
+  upstream/whisper-translate
+  upstream/wide_ep_working_branch
+  upstream/wide_ep_working_branch_2
+  upstream/woosuk-jf
+  upstream/woosuk/cleanup-flashinfer
+  upstream/woosuk/fa3-ima
+  upstream/woosuk/fa3-swa-cudagraph
+  upstream/woosuk/fix-graph-pool
+  upstream/woosuk/flashinfer-swa
+  upstream/woosuk/gumbel-sample
+  upstream/woosuk/input-prep
+  upstream/woosuk/minor-worker-fix
+  upstream/woosuk/model-runner-v2
+  upstream/woosuk/optimize-block-append
+  upstream/woosuk/remove-req-idx-mapping
+  upstream/woosuk/sampled-token-ids
+  upstream/woosuk/simple-cuda-graph
+  upstream/wye-refactor-w8a8-quant

tests/kernels/mamba/test_mamba_ssm_ssd.py

@@ -527,19 +527,18 @@ def test_mamba_chunk_scan_cont_batch_prefill_chunking(chunk_size, seqlens):
             Y_ref_seq[:chunked_seqlens[i], ...],
             atol=atol,
             rtol=rtol,
-            msg=lambda x: f"seq{i} output part1 " + x)  # noqa: B023
+            msg=lambda x, i=i: f"seq{i} output part1 " + x)
         torch.testing.assert_close(
             Y_seq[chunked_seqlens[i]:, ...],
             Y_ref_seq[chunked_seqlens[i]:, ...],
             atol=atol,
             rtol=rtol,
-            msg=lambda x: f"seq{i} output part2 " + x)  # noqa: B023
+            msg=lambda x, i=i: f"seq{i} output part2 " + x)
 
         state_seq = state_chunked[i]
         state_seq_ref = state_ref[i]
-        torch.testing.assert_close(
-            state_seq,
-            state_seq_ref,
-            atol=atol,
-            rtol=rtol,
-            msg=lambda x: f"seq{i} state " + x)  # noqa: B023
+        torch.testing.assert_close(state_seq,
+                                   state_seq_ref,
+                                   atol=atol,
+                                   rtol=rtol,
+                                   msg=lambda x, i=i: f"seq{i} state " + x)