[CI] Skip some flaky subtests for test_multi_lora_backend.py (#18408)

This commit is contained in:
Baizhou Zhang
2026-02-07 19:06:53 +08:00
committed by GitHub
parent c47c2f9466
commit eb4cf1dfc4
+22 -21
View File
@@ -548,7 +548,6 @@ def ensure_reproducibility():
random.seed(seed) random.seed(seed)
torch.manual_seed(seed) torch.manual_seed(seed)
torch.cuda.manual_seed_all(seed) torch.cuda.manual_seed_all(seed)
torch.use_deterministic_algorithms(True)
TEST_MULTIPLE_BATCH_PROMPTS = [ TEST_MULTIPLE_BATCH_PROMPTS = [
@@ -594,18 +593,19 @@ def create_multiple_batch_test_samples(
lora_adapter_paths[1], lora_adapter_paths[1],
], ],
), ),
( # It can pass half the time on CI, so skip this flaky case for now
[ # (
random.choice(prompts), # [
random.choice(prompts), # random.choice(prompts),
random.choice(prompts), # random.choice(prompts),
], # random.choice(prompts),
[ # ],
lora_adapter_paths[0], # [
None, # lora_adapter_paths[0],
lora_adapter_paths[1], # None,
], # lora_adapter_paths[1],
), # ],
# ),
( (
[ [
random.choice(prompts), random.choice(prompts),
@@ -614,14 +614,15 @@ def create_multiple_batch_test_samples(
], ],
[lora_adapter_paths[0], lora_adapter_paths[1], None], [lora_adapter_paths[0], lora_adapter_paths[1], None],
), ),
( # It can pass half the time on CI, so skip this flaky case for now
[ # (
random.choice(prompts), # [
random.choice(prompts), # random.choice(prompts),
random.choice(prompts), # random.choice(prompts),
], # random.choice(prompts),
[None, lora_adapter_paths[1], None], # ],
), # [None, lora_adapter_paths[1], None],
# ),
( (
[ [
random.choice(prompts), random.choice(prompts),