Apply isort and black reformatting

Signed-off-by: BoxiangW <[email protected]>

tests/collections/llm/hf/sft_fsdp2.py

@@ -15,13 +15,12 @@
 import fiddle as fdl
 import torch
 from lightning.pytorch.loggers import WandbLogger
+from packaging.version import Version as PkgVersion
+from utils import get_torch_version_str
 
 from nemo import lightning as nl
 from nemo.collections import llm
 from nemo.lightning.pytorch.accelerate.transformer_engine import is_te_accelerated
-from utils import get_torch_version_str
-
-from packaging.version import Version as PkgVersion
 
 DATA_PATH = '/home/TestData/lite/hf_cache/squad/'
 

tests/collections/llm/hf/sft_nemorun_fsdp2.py

@@ -13,16 +13,14 @@
 # limitations under the License.
 
 import nemo_run as run
+from packaging.version import Version as PkgVersion
+from utils import get_torch_version_str
 
 import nemo.lightning as nl
 from nemo.collections import llm
 from nemo.collections.common.tokenizers.huggingface.auto_tokenizer import AutoTokenizer
 from nemo.collections.llm.gpt.data.hf_dataset import SquadHFDataModule
 
-from utils import get_torch_version_str
-
-from packaging.version import Version as PkgVersion
-
 DATA_PATH = '/lustre/fsw/coreai_dlalgo_llm/boxiangw/squad'
 
 

tests/collections/llm/hf/utils.py

@@ -8,4 +8,4 @@ def get_torch_version_str():
     if hasattr(torch, '__version__'):
         return str(torch.__version__)
     else:
-        return version("torch")
+        return version("torch")