revert

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

Author: yzhang123

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

@@ -32,7 +32,6 @@
 from bionemo.testing.subprocess_utils import run_command_in_subprocess
 
 
-
 def small_training_llama_cmd(path, max_steps, val_check, devices: int = 1, additional_args: str = ""):
     cmd = (
         f"train_evo2 --no-fp32-residual-connection --mock-data --result-dir {path} --devices {devices} "
@@ -43,6 +42,7 @@ def small_training_llama_cmd(path, max_steps, val_check, devices: int = 1, addit
     )
     return cmd
 
+
 def test_predict_evo2_runs(
     tmp_path, num_sequences: int = 5, target_sequence_lengths: list[int] = [3149, 3140, 1024, 3149, 3149]
 ):
@@ -202,5 +202,3 @@ def test_predict_evo2_llama_runs(
         idx = seq_idx_map[seq_name]  # look up the out of order prediction index for this sequence.
         assert preds["pad_mask"][idx].sum() == expected_len
         assert preds["token_logits"][idx].shape == (max(target_sequence_lengths), 512)
-
-