Skip to content

Commit 033c202

Browse files
additional test case
1 parent 5a6b36f commit 033c202

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

tests/standalone_command_gen_strategy/test_aiconfigurator_standalone_command_gen_strategy.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,3 +129,26 @@ def test_cmd_args_requires_exactly_one_mode() -> None:
129129
isl=4000,
130130
osl=500,
131131
)
132+
133+
134+
def test_cmd_args_rejects_both_agg_and_disagg() -> None:
135+
with pytest.raises(ValueError):
136+
AiconfiguratorCmdArgs(
137+
model_name="LLAMA3.1_70B",
138+
system="h200_sxm",
139+
isl=4000,
140+
osl=500,
141+
agg=Agg(batch_size=8, ctx_tokens=16, tp=1, pp=1, dp=1, moe_tp=1, moe_ep=1),
142+
disagg=Disagg(
143+
p_tp=1,
144+
p_pp=1,
145+
p_dp=1,
146+
p_bs=1,
147+
p_workers=1,
148+
d_tp=1,
149+
d_pp=1,
150+
d_dp=1,
151+
d_bs=8,
152+
d_workers=2,
153+
),
154+
)

0 commit comments

Comments
 (0)