lint

Signed-off-by: Jared Wilber <jwilber@nvidia.com>

Author: jwilber

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

@@ -651,7 +651,7 @@ def train(args: argparse.Namespace) -> nl.Trainer:
         TimingCallback(),
         TEVCallback(),
     ]
-    
+
     if args.garbage_collect_at_inference:
         callbacks.append(GarbageCollectAtInferenceTime())
 

sub-packages/bionemo-evo2/src/bionemo/evo2/utils/logging/callbacks.py

@@ -43,7 +43,6 @@ class TEVCallback(Callback):
     - Data Parallelism: Only logs on rank 0 of each model parallel group
     """
 
-
     @torch.no_grad()
     def on_before_optimizer_step(self, trainer, pl_module, optimizer) -> None:
         """Called before each optimizer step during training.