Apply isort and black reformatting

Signed-off-by: huvunvidia <huvunvidia@users.noreply.github.com>
NVIDIA · Nov 15, 2024 · d8b41a0 · d8b41a0
1 parent fe541ed
commit d8b41a0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/nemo/collections/common/tokenizers/huggingface/auto_tokenizer.py b/nemo/collections/common/tokenizers/huggingface/auto_tokenizer.py
@@ -107,9 +107,11 @@ def __init__(
             print("use_fast: ", use_fast)
             print("trust_remote_code: ", trust_remote_code)
             import os
+
             print("os.path.exists(vocab_file): ", os.path.exists(vocab_file))
             try:
                 import subprocess
+
                 result = subprocess.run(['ls ' + vocab_file], capture_output=True, text=True)
                 print("Running ls command: ")
                 print(result.stdout)

diff --git a/nemo/collections/llm/__init__.py b/nemo/collections/llm/__init__.py
@@ -27,12 +27,6 @@
     PreTrainingDataModule,
     SquadDataModule,
 )
-from nemo.collections.llm.t5.data import (
-    FineTuningDataModule as T5FineTuningDataModule,
-    MockDataModule as T5MockDataModule,
-    PreTrainingDataModule as T5PreTrainingDataModule,
-    SquadDataModule as T5SquadDataModule,
-)
 from nemo.collections.llm.gpt.data.api import dolly, hf_dataset, mock, squad
 from nemo.collections.llm.gpt.model import (
     Baichuan2Config,
@@ -118,6 +112,10 @@
     gpt_forward_step,
 )
 from nemo.collections.llm.quantization import Quantizer, get_calib_data_iter
+from nemo.collections.llm.t5.data import FineTuningDataModule as T5FineTuningDataModule
+from nemo.collections.llm.t5.data import MockDataModule as T5MockDataModule
+from nemo.collections.llm.t5.data import PreTrainingDataModule as T5PreTrainingDataModule
+from nemo.collections.llm.t5.data import SquadDataModule as T5SquadDataModule
 from nemo.collections.llm.t5.model import T5Config, T5Model, t5_data_step, t5_forward_step
 
 __all__ = [

diff --git a/nemo/collections/llm/t5/data/__init__.py b/nemo/collections/llm/t5/data/__init__.py
@@ -1,6 +1,6 @@
 from nemo.collections.llm.t5.data.fine_tuning import FineTuningDataModule
+from nemo.collections.llm.t5.data.mock import MockDataModule
 from nemo.collections.llm.t5.data.pre_training import PreTrainingDataModule
 from nemo.collections.llm.t5.data.squad import SquadDataModule
-from nemo.collections.llm.t5.data.mock import MockDataModule
 
 __all__ = ["FineTuningDataModule", "PreTrainingDataModule", "SquadDataModule", "MockDataModule"]