@@ -401,7 +401,6 @@ def test_sanity_convergence_fsdp2_thd_fp8(tmp_path, recipe_path):
401401
402402
403403def test_sanity_ddp_thd (tmp_path , recipe_path ):
404-
405404 # For DDP, we only check that the script can run successfully with THD, not convergence.
406405 with initialize_config_dir (config_dir = str (recipe_path / "hydra_config" ), version_base = "1.2" ):
407406 sanity_config = compose (
@@ -418,7 +417,6 @@ def test_sanity_ddp_thd(tmp_path, recipe_path):
418417
419418
420419def test_sanity_mfsdp_thd (tmp_path , recipe_path ):
421-
422420 # For MFSDP, we only check that the script can run successfully with THD, not convergence.
423421 with initialize_config_dir (config_dir = str (recipe_path / "hydra_config" ), version_base = "1.2" ):
424422 sanity_config = compose (
@@ -436,7 +434,6 @@ def test_sanity_mfsdp_thd(tmp_path, recipe_path):
436434
437435@requires_fp8
438436def test_sanity_ddp_thd_fp8 (tmp_path , recipe_path ):
439-
440437 # For DDP, we only check that the script can run successfully with THD, not convergence.
441438 with initialize_config_dir (config_dir = str (recipe_path / "hydra_config" ), version_base = "1.2" ):
442439 sanity_config = compose (
@@ -455,7 +452,6 @@ def test_sanity_ddp_thd_fp8(tmp_path, recipe_path):
455452
456453@requires_fp8
457454def test_sanity_mfsdp_thd_fp8 (tmp_path , recipe_path ):
458-
459455 # For MFSDP, we only check that the script can run successfully with THD, not convergence.
460456 with initialize_config_dir (config_dir = str (recipe_path / "hydra_config" ), version_base = "1.2" ):
461457 sanity_config = compose (
@@ -548,7 +544,6 @@ def test_sanity_convergence_fsdp2_huggingface_model(tmp_path, recipe_path):
548544
549545
550546def test_sanity_ddp_thd_token_packing (tmp_path , recipe_path ):
551-
552547 # For DDP, we only check that the script can run successfully with THD, not convergence.
553548 with initialize_config_dir (config_dir = str (recipe_path / "hydra_config" ), version_base = "1.2" ):
554549 sanity_config = compose (
@@ -565,7 +560,6 @@ def test_sanity_ddp_thd_token_packing(tmp_path, recipe_path):
565560
566561
567562def test_sanity_mfsdp_thd_token_packing (tmp_path , recipe_path ):
568-
569563 with initialize_config_dir (config_dir = str (recipe_path / "hydra_config" ), version_base = "1.2" ):
570564 sanity_config = compose (
571565 config_name = "L0_sanity" ,
@@ -581,7 +575,6 @@ def test_sanity_mfsdp_thd_token_packing(tmp_path, recipe_path):
581575
582576
583577def test_sanity_fsdp2_thd_token_packing (tmp_path , recipe_path ):
584-
585578 with initialize_config_dir (config_dir = str (recipe_path / "hydra_config" ), version_base = "1.2" ):
586579 sanity_config = compose (
587580 config_name = "L0_sanity" ,
0 commit comments