style: fix lint issues in token_dtype_code fix

- Remove unused pytest import from new unit test

- Add missing blank line between top-level functions

Signed-off-by: andrewwhitecdw <andrewwhitecdw@users.noreply.github.com>

Author: andrewwhitecdw

src/megatron/bridge/training/setup.py

@@ -788,6 +788,7 @@ def _get_token_dtype_code(vocab_size: int) -> int:
 
     return 4 if vocab_size <= np.iinfo(np.uint16).max + 1 else 8
 
+
 def maybe_log_and_save_config(cfg: ConfigContainer) -> None:
     """Save configuration to disk and log non-default values on rank 0.
 

tests/unit_tests/bridge/training/test_setup.py

@@ -1,5 +1,4 @@
 import numpy as np
-import pytest
 
 from megatron.bridge.training.setup import _get_token_dtype_code