Skip to content

Commit af434ba

Browse files
committed
fix: remove extra blank lines after function defs in test_train.py
Pre-commit (ruff format) flagged 7 functions with a blank line between the def signature and the first body line. Removed them. Signed-off-by: svc-bionemo <267129667+svc-bionemo@users.noreply.github.com>
1 parent bdeba12 commit af434ba

1 file changed

Lines changed: 0 additions & 7 deletions

File tree

bionemo-recipes/recipes/esm2_native_te/tests/test_train.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -401,7 +401,6 @@ def test_sanity_convergence_fsdp2_thd_fp8(tmp_path, recipe_path):
401401

402402

403403
def 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

420419
def 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
438436
def 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
457454
def 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

550546
def 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

567562
def 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

583577
def 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

Comments
 (0)