revertin change to test_ad_build_small_single after rebase

govind-ramnarayan · govind-ramnarayan · commit 907222ca3978 · 2025-10-23T11:56:04.000-07:00
Signed-off-by: Govind Ramnarayan &lt;105831528+govind-ramnarayan@users.noreply.github.com&gt;
diff --git a/tests/unittest/_torch/auto_deploy/unit/singlegpu/test_ad_build_small_single.py b/tests/unittest/_torch/auto_deploy/unit/singlegpu/test_ad_build_small_single.py
@@ -28,7 +28,6 @@ def _check_ad_config(experiment_config: ExperimentConfig, llm_args: LlmArgs):
     expected_parallel_config = _ParallelConfig(
         tp_size=world_size, gpus_per_node=expected_llm_args.gpus_per_node
     )
-    expected_parallel_config.world_size = world_size
     assert llm_args._parallel_config == expected_parallel_config, (
         f"Expected parallel_config {expected_parallel_config}, got {llm_args._parallel_config}"
     )

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,6 @@ def _check_ad_config(experiment_config: ExperimentConfig, llm_args: LlmArgs):`
`28`	`28`	`expected_parallel_config = _ParallelConfig(`
`29`	`29`	`tp_size=world_size, gpus_per_node=expected_llm_args.gpus_per_node`
`30`	`30`	`)`
`31`		`- expected_parallel_config.world_size = world_size`
`32`	`31`	`assert llm_args._parallel_config == expected_parallel_config, (`
`33`	`32`	`f"Expected parallel_config {expected_parallel_config}, got {llm_args._parallel_config}"`
`34`	`33`	`)`