fix precommit

Signed-off-by: Yang Zhang <yangzhang@nvidia.com>

Author: yzhang123

sub-packages/bionemo-evo2/src/bionemo/evo2/data/sharded_eden_dataloader.py

@@ -100,7 +100,7 @@ def __init__(
         log_dir: Optional[str] = None,
         **kwargs,
     ):
-        """Initialize the ShardedEdenDataModule. See sub-packages/bionemo-evo2/src/bionemo/evo2/data/sharded_eden_dataloader.md for how to prepare the input data. """
+        """Initialize the ShardedEdenDataModule. See sub-packages/bionemo-evo2/src/bionemo/evo2/data/sharded_eden_dataloader.md for how to prepare the input data."""
         super().__init__()
         self.sequence_db_dir = sequence_db_dir
         self.train_window_db_path = train_window_db_path

sub-packages/bionemo-evo2/src/bionemo/evo2/run/train.py

@@ -1070,7 +1070,9 @@ def train(args: argparse.Namespace) -> nl.Trainer:
         constant_steps=args.constant_steps,
     )
     # This is where the no weight decay condition is applied to the optimizer state.
-    opt = MegatronOptimizerModule(opt_config, sched, no_weight_decay_cond=getattr(model_config, 'hyena_no_weight_decay_cond_fn', None))
+    opt = MegatronOptimizerModule(
+        opt_config, sched, no_weight_decay_cond=getattr(model_config, "hyena_no_weight_decay_cond_fn", None)
+    )
     opt.connect(model)
     # Start training
     trainer.fit(model, data_module)

sub-packages/bionemo-evo2/tests/bionemo/evo2/data/test_sharded_eden_dataset.py

@@ -480,8 +480,6 @@ def test_dataset_collate_fn(sequence_db_dir, window_dbs):
     dataset.__del__()
 
 
-
-
 def test_window_min_length_threshold(temp_dir, train_parquet):
     """Test window database creation with length threshold."""
     output_db = Path(temp_dir) / "threshold_windows.db"
@@ -545,4 +543,3 @@ def test_dataset_length_and_iteration(sequence_db_dir, window_dbs):
 
     # Clean up
     dataset.__del__()
-

sub-packages/bionemo-evo2/tests/bionemo/evo2/run/test_train.py

@@ -119,6 +119,7 @@ def small_training_llama_finetune_cmd(
     )
     return cmd
 
+
 @pytest.mark.timeout(512)  # Optional: fail if the test takes too long.
 @pytest.mark.slow
 def test_train_evo2_finetune_runs(tmp_path):
@@ -266,7 +267,6 @@ def test_train_evo2_mamba_finetune_runs(tmp_path):
     assert len(matching_subfolders_ft) == 1, "Only one checkpoint subfolder should be found."
 
 
-
 @pytest.mark.timeout(512)  # Optional: fail if the test takes too long.
 @pytest.mark.slow
 def test_train_evo2_llama_finetune_runs(tmp_path):